在 Slurm 中，是否可以为每个任务分配不同数量的 CPU？答案

【问题标题】：In Slurm, is it possible to assign a different number of CPUs for every task?在 Slurm 中，是否可以为每个任务分配不同数量的 CPU？
【发布时间】：2020-12-13 05:47:47
【问题描述】：

我在 Slurm 集群中运行 MPI-over-openmp 作业，我正在尝试找出一种方法，为每个生成的任务分配不同数量的 CPU。例如，假设我们运行这个作业：

srun --nodes 1 --ntasks 2 --cpus-per-task 2 ./mpi_exe

这将在单个节点中生成 2 个 MPI 进程，每个进程有 2 个 CPU。但是，例如，我想为第一个进程分配 3 个 CPU，在第二个进程中分配 1 个。

你知道有什么方法可以实现吗？

【问题讨论】：

标签： mpi slurm

【解决方案1】：

看看Heterogeneous Jobs。对于您的示例，这应该可以解决问题：

srun -N1 -n1 -c3 : -N1 -n1 -c1 ./mpi_exe

【讨论】：