下面是一个提交作业的脚本示例:

#!/bin/bash
#SBATCH --job-name=myjob
#SBATCH --nodes=4
#SBATCH --ntasks-per-node=16
#SBATCH --cpus-per-task=1
#SBATCH --time=00:10:00

module load intelmpi

srun --mpi=pmi2 ./myprogram

解释如下:

  • --job-name: 指定作业名称。
  • --nodes: 指定使用的节点数。
  • --ntasks-per-node: 指定每个节点上使用的进程数,这里是16个。
  • --cpus-per-task: 指定每个进程使用的CPU核数,这里是1个,因为MPI会自动将进程分配到空闲的核上。
  • --time: 指定作业最长运行时间,这里是10分钟。

srun命令中,--mpi=pmi2参数指定使用PMI2接口,这是MPI的一种启动方式。./myprogram是要运行的程序。MPI会自动将进程分配到每个节点上,每个节点上使用16个核来运行程序。

使用Slurm调度系统提交多节点MPI作业:每个节点16核,总计64核

原文地址: https://www.cveoy.top/t/topic/oghO 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录