hailin
/
train_env_prepare
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

.

This commit is contained in:
hailin 2025-08-08 21:31:32 +08:00
parent 712b3bb704
commit 504617890c
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
dspeed/install_deepspeed_src_mamba_single_gpu.sh
View File

@ -5,7 +5,7 @@ set -euo pipefail
# Config（可用环境变量覆盖） # Config（可用环境变量覆盖）
# ============================= # =============================
REPO_URL="${REPO_URL:-https://github.com/microsoft/DeepSpeed.git}" 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”下操作 REPO_DIR="${REPO_DIR:-DeepSpeed}" # 将在“当前目录/REPO_DIR”下操作
CUDA_VERSION_PIN="${CUDA_VERSION_PIN:-11.8}" # 期望 Torch 的 CUDA tag 和 nvcc 都是 11.8 CUDA_VERSION_PIN="${CUDA_VERSION_PIN:-11.8}" # 期望 Torch 的 CUDA tag 和 nvcc 都是 11.8
ARCH_LIST="${TORCH_CUDA_ARCH_LIST:-}" # 为空则自动探测 ARCH_LIST="${TORCH_CUDA_ARCH_LIST:-}" # 为空则自动探测