如何在程序中获取SLURM任务ID

Question

我在跑srun -n 100 python foo.py。在 python 脚本中，它如何找出它是哪个任务编号/id/等级？是否有设置环境变量？

Answer 1

查看man srun或man sbatch获取环境变量列表。$SLURM_PROCID可能就是您需要的。