Fix installation for nvidia-nvshmem-cu12 (#14033)

2025-11-26 18:27:12 -08:00
parent e9e90460ed
commit b12c9e5c0a
2 changed files with 4 additions and 3 deletions
@@ -134,6 +134,10 @@ if [ "$IS_BLACKWELL" != "1" ]; then
    git clone --branch v0.5 --depth 1 https://github.com/EvolvingLMMs-Lab/lmms-eval.git
    $PIP_CMD install -e lmms-eval/ $PIP_INSTALL_SUFFIX
 fi
+
+# DeepEP depends on nvshmem 3.4.5
+$PIP_CMD install nvidia-nvshmem-cu12==3.4.5 --force-reinstall $PIP_INSTALL_SUFFIX
+
 # Cudnn with version less than 9.16.0.29 will cause performance regression on Conv3D kernel
 $PIP_CMD install nvidia-cudnn-cu12==9.16.0.29 --force-reinstall $PIP_INSTALL_SUFFIX
 $PIP_CMD uninstall xformers || true