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

Question

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

Answer 1

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