mirror of
https://github.com/modelscope/modelscope.git
synced 2025-12-25 04:29:22 +01:00
04b9704b48
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/11371422 * v1.1 docker file modify
7 lines
255 B
Bash
7 lines
255 B
Bash
export MAX_JOBS=16
|
|
git clone https://github.com/NVIDIA/apex
|
|
cd apex
|
|
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" ./
|
|
cd ..
|
|
rm -rf apex
|