gstykalin/test-dags
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update dags

Browse Source
This commit is contained in:
George Stykalin
2026-04-15 11:07:47 +03:00
parent b3b527f568
commit 276c1ba6c5
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
dags/test-train-pytorch.py
View File

@@ -31,6 +31,7 @@ def run_rank_0(**context):
"MASTER_ADDR": fqdn, "MASTER_ADDR": fqdn,
"MASTER_PORT": MASTER_PORT, "MASTER_PORT": MASTER_PORT,
"WORLD_SIZE": "2", "WORLD_SIZE": "2",
"NCCL_IB_GID_INDEX": "0",
"RANK": "0", "RANK": "0",
"NCCL_SOCKET_IFNAME": "eth0", # bootstrap/rendezvous через eth "NCCL_SOCKET_IFNAME": "eth0", # bootstrap/rendezvous через eth
"NCCL_IB_DISABLE": "0", # трафик через IB "NCCL_IB_DISABLE": "0", # трафик через IB