在具有指定ID的CPU上运行mpirun

Question

有人知道如何在指定的CPU上执行mpirun吗？虽然“mpiunCPU4”指定了所使用的-np的数量，但我在这里要做的是指定CPU。

操作系统为CentOS 5.6，在具有6x2内核的单个节点上使用MVAPICH2。

谢谢合作。

Answer 1

可以；新版本的mvapich2使用hwloc库来启用CPU亲和性和绑定。

从User Guide

例如，如果您想在每个节点上运行4个进程，并在每个节点上使用内核0、1、4、5，则可以指定：

$ mpirun_rsh -np 64 -hostfile hosts MV2_CPU_MAPPING=0:1:4:5 ./a.out

或

$ mpiexec -n 64 -f hosts -env MV2_CPU_MAPPING 0:1:4:5 ./a.out

这样，每个节点上的进程0将映射到核0，进程1将映射到核1，进程2将映射到核4，进程3将映射到核5。对于每个进程，映射由单个“：”分隔。