From e68097db7766e50db81551dd93c57ce43cd2dce7 Mon Sep 17 00:00:00 2001 From: lingyuzeng Date: Sun, 14 Jul 2024 08:47:58 +0800 Subject: [PATCH] update download.pytorch.org/whl --- finetune/Dockerfile.update | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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