Как получить идентификатор задачи SLURM в программе

Question

Я бегу srun -n 100 python foo.py. Внутри скрипта Python как узнать, какой это номер задачи / id / rank? Установлена ​​ли переменная окружения?

Answer 1

Посмотрите на man srun или man sbatch список переменных среды. $SLURM_PROCID может быть тот, который вам нужен.