diff --git a/dspeed/install_deepspeed_src_mamba_single_gpu.sh b/dspeed/install_deepspeed_src_mamba_single_gpu.sh index f6919d1..92ba92e 100755 --- a/dspeed/install_deepspeed_src_mamba_single_gpu.sh +++ b/dspeed/install_deepspeed_src_mamba_single_gpu.sh @@ -221,6 +221,9 @@ export DS_BUILD_CPU_ADAM=1 # 可选:异步 IO(Linux 才生效;Windows 不支持 AIO/GDS) export DS_BUILD_AIO=1 +export DS_BUILD_INFERENCE=0 +export DS_BUILD_CUTLASS=0 + # 安装(带详细日志) pip install -v .