如何将ntasks值添加至Slurm/sbatch输出及错误文件名

在由Slurm管理的并行程序中，我使用sbatch的输出文件命名模式生成包含作业名和作业ID的输出文件。但我需要基于处理器数量分析程序运行时间，因此想把所用任务数添加到文件名里方便追踪。

目前我的MPI程序会打印运行的进程数，但要打开每个文件查看内容，不利于用Shell便捷处理。请问如何生成类似%J.%T.%j格式（点分隔作业名、任务数、作业ID）的文件名？

Slurm的sbatch本身没有直接对应任务数的占位符，但可以通过环境变量拼接出符合要求的文件名，以下是两种实用方法：

方法1：在sbatch脚本内定义 #

在你的批处理脚本开头，直接用Slurm提供的环境变量构建输出文件名：

#!/bin/bash
#SBATCH --job-name=my_parallel_job
#SBATCH --ntasks=16
# 按要求拼接文件名：作业名.任务数.作业ID
#SBATCH --output=${SLURM_JOB_NAME}.${SLURM_NTASKS}.%j.out
#SBATCH --error=${SLURM_JOB_NAME}.${SLURM_NTASKS}.%j.err

# 执行MPI程序
mpiexec ./your_mpi_executable

提交后会生成类似my_parallel_job.16.789012.out的文件，完全匹配你想要的格式。

方法2：提交命令行指定 #

如果不想修改脚本，提交时可以直接在命令行传递带变量的文件名参数：

sbatch --job-name=my_job --ntasks=8 --output='${SLURM_JOB_NAME}.${SLURM_NTASKS}.%j.out' your_script.sh

这里必须用单引号包裹文件名，防止Shell提前解析环境变量，确保Slurm能正确替换变量。

核心变量说明 #

• SLURM_JOB_NAME：对应作业名，和%J占位符作用相同
• SLURM_NTASKS：作业分配的总任务数，也就是你需要追踪的处理器/进程数量
• %j：Slurm内置的作业ID占位符，也可以用SLURM_JOB_ID替代

生成的文件名可以直接用ls、grep等Shell工具按任务数筛选，不用打开文件就能批量处理，非常适合后续的运行时间分析。

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