diff --git a/finetune/Dockerfile.update b/finetune/Dockerfile.update
index 6b8941d..3fc447b 100644
--- a/finetune/Dockerfile.update
+++ b/finetune/Dockerfile.update
@@ -103,7 +103,7 @@ python3 -m pip install --no-cache-dir ./transformers[deepspeed-testing]
 python3 -m pip uninstall -y torch torchvision torchaudio 
 # # install pytorch create conda env aleay exists
 # 直接将 PyTorch 安装指引 中的 https://download.pytorch.org/whl 替换为 https://mirror.sjtu.edu.cn/pytorch-wheels 即可。
-python3 -m pip install torch==${PYTORCH_VERSION}+${CUDA} torchvision==${TORCHVISION_VERSION}+${CUDA} torchaudio==${TORCHAUDIO_VERSION} xformers --extra-index-url https://mirror.sjtu.edu.cn/pytorch-wheels/${CUDA}
+python3 -m pip install torch==${PYTORCH_VERSION}+${CUDA} torchvision==${TORCHVISION_VERSION}+${CUDA} torchaudio==${TORCHAUDIO_VERSION} xformers --extra-index-url https://download.pytorch.org/whl/${CUDA}
 python3 -m pip install --no-cache-dir git+https://github.com/huggingface/accelerate@main#egg=accelerate
 python3 -m pip uninstall -y transformer-engine
 python3 -m pip uninstall -y torch-tensorrt