From 504617890cbfb2757828d039248bf3340ad71ce5 Mon Sep 17 00:00:00 2001 From: hailin Date: Fri, 8 Aug 2025 21:31:32 +0800 Subject: [PATCH] . --- dspeed/install_deepspeed_src_mamba_single_gpu.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dspeed/install_deepspeed_src_mamba_single_gpu.sh b/dspeed/install_deepspeed_src_mamba_single_gpu.sh index 9ff64f5..f9772e7 100644 --- a/dspeed/install_deepspeed_src_mamba_single_gpu.sh +++ b/dspeed/install_deepspeed_src_mamba_single_gpu.sh @@ -5,7 +5,7 @@ set -euo pipefail # Config（可用环境变量覆盖） # ============================= REPO_URL="${REPO_URL:-https://github.com/microsoft/DeepSpeed.git}" -REPO_BRANCH="${REPO_BRANCH:-v0.17.7}" +REPO_BRANCH="${REPO_BRANCH:-v0.17.4}" REPO_DIR="${REPO_DIR:-DeepSpeed}" # 将在“当前目录/REPO_DIR”下操作 CUDA_VERSION_PIN="${CUDA_VERSION_PIN:-11.8}" # 期望 Torch 的 CUDA tag 和 nvcc 都是 11.8 ARCH_LIST="${TORCH_CUDA_ARCH_LIST:-}" # 为空则自动探测