update dags

2026-04-15 11:07:47 +03:00
parent b3b527f568
commit 276c1ba6c5
1 changed files with 1 additions and 0 deletions
--- a/dags/test-train-pytorch.py
+++ b/dags/test-train-pytorch.py
@@ -31,6 +31,7 @@ def run_rank_0(**context):
        "MASTER_ADDR": fqdn,
        "MASTER_PORT": MASTER_PORT,
        "WORLD_SIZE": "2",
+        "NCCL_IB_GID_INDEX": "0",
        "RANK": "0",
        "NCCL_SOCKET_IFNAME": "eth0",   # bootstrap/rendezvous через eth
        "NCCL_IB_DISABLE": "0",         # трафик через IB