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

Question

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

Answer 1

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