From 8f3e78c16a8b2460521488ef4526b5e9f956f81d Mon Sep 17 00:00:00 2001 From: George Stykalin Date: Tue, 14 Apr 2026 23:58:12 +0300 Subject: [PATCH] update dags --- dags/test-train-pytorch.py | 1 + 1 file changed, 1 insertion(+) diff --git a/dags/test-train-pytorch.py b/dags/test-train-pytorch.py index 720a87f..497d4fd 100644 --- a/dags/test-train-pytorch.py +++ b/dags/test-train-pytorch.py @@ -84,6 +84,7 @@ def run_training_node_func(rank, world_size): os.environ['WORLD_SIZE'] = str(world_size) os.environ['RANK'] = str(rank) os.environ['NCCL_SOCKET_IFNAME'] = 'eth0' + os.environ['NCCL_PORT_RANGE'] = '30000-30100' os.environ['NCCL_DEBUG'] = 'INFO' os.environ['NCCL_TIMEOUT'] = str(NCCL_TIMEOUT) os.environ['NCCL_BLOCKING_WAIT'] = '1'