Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-02-24 04:01:10 +01:00
Code Issues Packages Projects Releases Wiki Activity

update deepspeed==0.17.4 due to the gpu image imports error

Browse Source
This commit is contained in:
班扬
2025-09-28 14:48:16 +08:00
parent 9b4cd6fe61
commit 655bef00f2
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
docker/build_image.py
View File

@@ -209,6 +209,7 @@ class GPUImageBuilder(Builder):
RUN pip install tf-keras==2.16.0 --no-dependencies && \
pip install onnx==1.18.0 --no-dependencies && \
pip install transformers==4.55.4 --no-dependencies && \
pip install deepspeed==0.17.4 --no-dependencies && \
pip install --no-cache-dir torchsde jupyterlab torchmetrics==0.11.4 basicsr pynvml shortuuid && \
CUDA_HOME=/usr/local/cuda TORCH_CUDA_ARCH_LIST="6.0 6.1 7.0 7.5 8.0 8.6 8.9 9.0" \
pip install --no-cache-dir 'git+https://github.com/facebookresearch/detectron2.git'