Update train_deepspeed.sh

2025-07-11 18:32:16 +08:00
parent 8eff6d8233
commit f4e0307665
1 changed files with 1 additions and 2 deletions
--- a/hy3dshape/scripts/train_deepspeed.sh
+++ b/hy3dshape/scripts/train_deepspeed.sh
@@ -29,10 +29,9 @@ else
    export NCCL_IB_SL=3
    export NCCL_CHECK_DISABLE=1
    export NCCL_P2P_DISABLE=0
-    export NCCL_IB_DISABLE=1
+    export NCCL_IB_DISABLE=0
    export NCCL_LL_THRESHOLD=16384
    export NCCL_IB_CUDA_SUPPORT=1
-    # DELETED: The hardcoded export NCCL_SOCKET_IFNAME and UCX_NET_DEVICES lines
    export NCCL_COLLNET_ENABLE=0
    export SHARP_COLL_ENABLE_SAT=0
    export NCCL_NET_GDR_LEVEL=2