From b07455fac45e29a1de1cb2ba912621666cb9387e Mon Sep 17 00:00:00 2001 From: "yuze.zyz" Date: Tue, 22 Oct 2024 17:57:59 +0800 Subject: [PATCH] fix --- docker/Dockerfile.ubuntu_base | 2 ++ docker/install.sh | 8 ++++++-- docker/install_cpu.sh | 2 +- 3 files changed, 9 insertions(+), 3 deletions(-) diff --git a/docker/Dockerfile.ubuntu_base b/docker/Dockerfile.ubuntu_base index ea781504..19904b86 100644 --- a/docker/Dockerfile.ubuntu_base +++ b/docker/Dockerfile.ubuntu_base @@ -210,4 +210,6 @@ ARG TENSORFLOW_VERSION={tf_version} echo 'cpu not install tinycudann'; \ fi + RUN pip install --no-cache-dir fairseq + ENTRYPOINT [] diff --git a/docker/install.sh b/docker/install.sh index 28bfd6bb..36bd1a19 100644 --- a/docker/install.sh +++ b/docker/install.sh @@ -11,7 +11,7 @@ swift_branch=${8:-main} pip uninstall -y torch torchvision torchaudio -pip install --no-cache-dir -U torch==$torch_version torchvision==$torchvision_version torchaudio==$torchaudio_version +pip install --no-cache-dir torch==$torch_version torchvision==$torchvision_version torchaudio==$torchaudio_version pip install --no-cache-dir funtextprocessing typeguard==2.13.3 scikit-learn -f https://modelscope.oss-cn-beijing.aliyuncs.com/releases/repo.html @@ -25,7 +25,11 @@ pip install --no-cache-dir torchsde jupyterlab torchmetrics==0.11.4 tiktoken tra pip install --no-cache-dir text2sql_lgesql==1.3.0 git+https://github.com/jin-s13/xtcocoapi.git@v1.14 git+https://github.com/gatagat/lap.git@v0.4.0 -f https://modelscope.oss-cn-beijing.aliyuncs.com/releases/repo.html --force --no-deps -pip install --no-cache-dir mpi4py paint_ldm mmcls>=0.21.0 mmdet>=2.25.0 decord>=0.6.0 ipykernel fasttext fairseq deepspeed apex -f https://modelscope.oss-cn-beijing.aliyuncs.com/releases/repo.html +pip install --no-cache-dir mpi4py paint_ldm -f https://modelscope.oss-cn-beijing.aliyuncs.com/releases/repo.html + +pip install --no-cache-dir mmcls>=0.21.0 mmdet>=2.25.0 decord>=0.6.0 -f https://modelscope.oss-cn-beijing.aliyuncs.com/releases/repo.html + +pip install --no-cache-dir ipykernel fasttext deepspeed apex -f https://modelscope.oss-cn-beijing.aliyuncs.com/releases/repo.html 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'; diff --git a/docker/install_cpu.sh b/docker/install_cpu.sh index 4fa3ce38..d8ad6c95 100644 --- a/docker/install_cpu.sh +++ b/docker/install_cpu.sh @@ -20,7 +20,7 @@ pip install --no-cache-dir -U qwen_vl_utils pyav librosa timm transformers accel pip install --no-cache-dir text2sql_lgesql==1.3.0 git+https://github.com/jin-s13/xtcocoapi.git@v1.14 git+https://github.com/gatagat/lap.git@v0.4.0 -f https://modelscope.oss-cn-beijing.aliyuncs.com/releases/repo.html --force --no-deps -pip install --no-cache-dir mpi4py paint_ldm mmcls>=0.21.0 mmdet>=2.25.0 decord>=0.6.0 ipykernel fasttext fairseq -f https://modelscope.oss-cn-beijing.aliyuncs.com/releases/repo.html +pip install --no-cache-dir mpi4py paint_ldm mmcls>=0.21.0 mmdet>=2.25.0 decord>=0.6.0 ipykernel fasttext -f https://modelscope.oss-cn-beijing.aliyuncs.com/releases/repo.html pip uninstall ms-swift modelscope -y