Slurm技术问题：如何使用节点的全部可用核心？

针对不同分区节点核心数不一致的问题，Slurm提供了无需硬编码核心数的解决方案：

1. 任务并行场景（每个任务占用1核心） #

直接使用--ntasks-per-node=ALL参数，Slurm会自动识别所分配节点的全部可用核心数量，为每个核心分配一个任务。示例提交脚本片段：

#SBATCH --ntasks-per-node=ALL
# 其他SBATCH参数（如分区、时间等）
srun ./your_parallel_program

2. 单任务多线程场景（如OpenMP程序） #

先通过--exclusive参数独占整个节点，再利用Slurm自动设置的环境变量$SLURM_CPUS_ON_NODE获取当前节点的核心数，以此配置线程数。示例：

#SBATCH --exclusive
# 其他SBATCH参数
export OMP_NUM_THREADS=$SLURM_CPUS_ON_NODE
./your_multithreaded_program

补充说明 #

• --ntasks-per-node=ALL在多数Slurm版本中都支持，会自动匹配节点核心总数；
• --exclusive确保节点资源完全分配给你的作业，避免其他作业共享核心；
• 环境变量$SLURM_CPUS_ON_NODE会在作业分配后自动填充为当前节点的可用核心数，无需手动设置。

内容的提问来源于stack exchange，提问作者Heatherosa