Adding HYDRAGNN_MASTER_PORT

Adding HYDRAGNN_MASTER_ADDR env to set custom DDP port.
ORNL · Jul 3, 2024 · 940a1fe · 940a1fe
1 parent 6d6ff8d
commit 940a1fe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hydragnn/utils/distributed.py b/hydragnn/utils/distributed.py
@@ -130,7 +130,7 @@ def setup_ddp():
 
     ## Default setting
     master_addr = "127.0.0.1"
-    master_port = "8889"
+    master_port = os.getenv("HYDRAGNN_MASTER_PORT", "8889")
 
     if os.getenv("HYDRAGNN_MASTER_ADDR") is not None:
         master_addr = os.environ["HYDRAGNN_MASTER_ADDR"]