Slurm:最大变量 SLURM_ARRAY_TASK_ID

Slurm: variable for max SLURM_ARRAY_TASK_ID

我有一个简单的 slurm 作业文件,如下所示:

#!/bin/bash

#SBATCH --array=1-1000
#SBATCH -t 60:00
#SBATCH --mail-type=ALL
python cats.py ${SLURM_ARRAY_TASK_ID} 1000

第二个参数是为了让我的脚本知道这项工作的工人总数。

不过,我想将 1000 值变成一个变量,这样我就不需要对工人总数进行硬编码。当前作业中是否有一些用于最大数组任务 ID 的 slurm 变量?

可以使用环境变量SLURM_ARRAY_TASK_MAX