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

Question

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

Answer 1

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