From 49d7fa3478daa7f4c61f84d975a4baf0d1504272 Mon Sep 17 00:00:00 2001
From: hailin <hailin@gdzx.xyz>
Date: Fri, 8 Aug 2025 22:18:55 +0800
Subject: [PATCH] .

---
 dspeed/install_deepspeed_src_mamba_single_gpu.sh | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/dspeed/install_deepspeed_src_mamba_single_gpu.sh b/dspeed/install_deepspeed_src_mamba_single_gpu.sh
index acccaf9..79fc48f 100755
--- a/dspeed/install_deepspeed_src_mamba_single_gpu.sh
+++ b/dspeed/install_deepspeed_src_mamba_single_gpu.sh
@@ -166,6 +166,21 @@ fi
 # =============================
 log "Build & install DeepSpeed (training only, no inference ops)"
 
+
+# 先把 libcurand 的开发包装上（提供无版本名 libcurand.so）
+set +e
+${CONDA_BIN} install -y -c "nvidia/label/cuda-${CUDA_VERSION_PIN}.0" libcurand-dev
+RC=$?
+set -e
+if [[ $RC -ne 0 ]]; then
+  log "nvidia channel 没装上 libcurand-dev，回退到 conda-forge"
+  ${CONDA_BIN} install -y -c conda-forge libcurand libcurand-dev
+fi
+
+# 链接器需要能找到这些库
+export LD_LIBRARY_PATH="${CONDA_PREFIX}/lib64:${CONDA_PREFIX}/lib:${LD_LIBRARY_PATH:-}"
+export LIBRARY_PATH="${CONDA_PREFIX}/lib64:${CONDA_PREFIX}/lib:${LIBRARY_PATH:-}"
+
 export DS_BUILD_INFERENCE=0
 export DS_BUILD_CUTLASS=0
 export DS_BUILD_QUANTIZER=0