ClearML/clearml-agent
1
0
Fork 0
mirror of https://github.com/clearml/clearml-agent synced 2025-01-31 00:56:53 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix slurm multi-node rank detection

Browse Source
This commit is contained in:
allegroai 2024-07-24 17:44:05 +03:00
parent cd5b4d2186
commit f18c6b809f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
clearml_agent/commands/worker.py
View File

@ -2086,7 +2086,7 @@ class Worker(ServiceCommandSection):
if lines and ENV_MULTI_NODE_SINGLE_TASK.get() and ENV_MULTI_NODE_SINGLE_TASK.get() > 0:
# noinspection PyBroadException
try:
rank = int(os.environ.get("RANK") or 0)
rank = int(os.environ.get("RANK", os.environ.get('SLURM_PROCID')) or 0)
except Exception:
rank = 0
if rank: