From 5e53b5e5a0511c09e5df430393b01209847bf5ca Mon Sep 17 00:00:00 2001 From: Jintao Huang Date: Mon, 11 Nov 2024 14:50:30 +0800 Subject: [PATCH] update llm docker --- docker/Dockerfile.ubuntu | 1 + docker/build_image.py | 1 - 2 files changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/Dockerfile.ubuntu b/docker/Dockerfile.ubuntu index 8431aad0..3461e32e 100644 --- a/docker/Dockerfile.ubuntu +++ b/docker/Dockerfile.ubuntu @@ -20,6 +20,7 @@ ARG INSTALL_MS_DEPS={install_ms_deps} COPY requirements /var/modelscope RUN pip uninstall ms-swift modelscope -y && pip --no-cache-dir install pip==23.* -U && \ + pip install --no-cache-dir apex -f https://modelscope.oss-cn-beijing.aliyuncs.com/releases/repo.html && \ if [ "$INSTALL_MS_DEPS" = "True" ]; then \ pip --no-cache-dir install omegaconf==2.0.6 && \ pip install --no-cache-dir 'cython<=0.29.36' versioneer 'numpy<2.0' -f https://modelscope.oss-cn-beijing.aliyuncs.com/releases/repo.html && \ diff --git a/docker/build_image.py b/docker/build_image.py index a9e1a107..e3fc930f 100644 --- a/docker/build_image.py +++ b/docker/build_image.py @@ -195,7 +195,6 @@ class GPUImageBuilder(Builder): RUN pip install adaseq pai-easycv && \ pip install tf-keras==2.16.0 --no-dependencies && \ pip install --no-cache-dir torchsde jupyterlab torchmetrics==0.11.4 basicsr pynvml shortuuid && \ - pip install --no-cache-dir 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' """