运行时是否有内置的方法来获取Slurm作业的ID?
我可以想象通过解析squeue并将ID与主机名进行匹配来确定ID(例如socket.gethostname()在python中)。有更好的解决方案吗?
socket.gethostname()
作业ID被分配给SLURM_JOB_ID环境变量。因此,在Python中,你将获得
SLURM_JOB_ID
import os print(os.environ["SLURM_JOB_ID"])
请注意,即使作业ID是整数,环境变量也包含字符串,因此你将需要调用int()函数将其转换。
int()