diff --git a/docker/Dockerfile.ubuntu_base b/docker/Dockerfile.ubuntu_base
index 1cccf754..b69f4127 100644
--- a/docker/Dockerfile.ubuntu_base
+++ b/docker/Dockerfile.ubuntu_base
@@ -145,7 +145,7 @@ ARG TORCH_VERSION={torch_version}
 ARG CUDATOOLKIT_VERSION={cudatoolkit_version}
 
 RUN if [ "$USE_GPU" = "True" ] ; then \
-        pip install --no-cache-dir torch==$TORCH_VERSION torchvision torchaudio; \
+        pip install --no-cache-dir torch==$TORCH_VERSION torchvision torchaudio -f https://modelscope.oss-cn-beijing.aliyuncs.com/releases/repo.html; \
     else \
         pip install --no-cache-dir torch==$TORCH_VERSION torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cpu; \
     fi
@@ -155,7 +155,7 @@ RUN if [ "$USE_GPU" = "True" ] ; then \
 ENV SHELL=/bin/bash
 # install special package
 RUN if [ "$USE_GPU" = "True" ] ; then \
-        pip install  dgl -f https://data.dgl.ai/wheels/$CUDATOOLKIT_VERSION/repo.html; \
+        pip install --no-cache-dir dgl -f https://data.dgl.ai/wheels/$CUDATOOLKIT_VERSION/repo.html; \
     else \
         pip install --no-cache-dir dgl dglgo -f https://data.dgl.ai/wheels/repo.html; \
     fi
diff --git a/docker/install.sh b/docker/install.sh
index a4dddf10..e92db1ef 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 -U torch==$torch_version torchvision==$torchvision_version torchaudio==$torchaudio_version -f https://modelscope.oss-cn-beijing.aliyuncs.com/releases/repo.html
 
 pip install --no-cache-dir funtextprocessing typeguard==2.13.3 scikit-learn -f https://modelscope.oss-cn-beijing.aliyuncs.com/releases/repo.html
 
@@ -39,11 +39,11 @@ pip install --no-cache-dir auto-gptq==$autogptq_version
 
 pip install --no-cache-dir -U triton
 
-pip install vllm==$vllm_version -U
+pip install --no-cache-dir vllm==$vllm_version -U
 
 pip install --no-cache-dir -U lmdeploy==$lmdeploy_version --no-deps
 
-pip install pynvml shortuuid
+pip install --no-cache-dir pynvml shortuuid
 
 pip uninstall ms-swift modelscope -y