modelscope
/
ModelScope
Not watched
1
0
Fork 0
Code
Releases 0 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Tree: b0cf09d7b0
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'b0cf09d7b0'
${ noResults }
ModelScope/docker/scripts/install_apex.sh

6 lines
236 B
Raw Blame History 

  1. git clone https://github.com/NVIDIA/apex

  2. cd apex

  3. TORCH_CUDA_ARCH_LIST="6.0;6.1;6.2;7.0;7.5;8.0;8.6" pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./

  4. cd ..

  5. rm -rf apex