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
@@ -46,9 +46,6 @@ if [ ! -e "$LIB_PATH/libmlx5.so" ]; then
 fi
 apt-get update && apt-get install -y libfabric-dev

-# Install NVSHMEM
-pip install nvidia-nvshmem-cu12==3.4.5 --force-reinstall
-
 # Install DeepEP
 DEEPEP_DIR=/root/.cache/deepep
 rm -rf ${DEEPEP_DIR}