如何通过mpirun在Linux系统中将Python进程绑定到指定的不同核心？

看起来你遇到的是OpenMPI进程核心绑定不严格的问题，这种情况我之前也碰到过，主要是参数组合没配对导致的。先帮你拆解下问题，再给你几个靠谱的解决方法。

问题根源分析 #

你用mpirun --cpu-set 15-20 --bind-to core -n 6 python3 bla.py时，出现核心重复绑定、部分核心空置的情况，而且--display-map显示进程都是UNBOUND——这说明OpenMPI并没有严格按照你预期的规则分配核心。

原因很简单：--cpu-set只是划定了可用核心的范围，但你没明确告诉它如何把每个进程一对一映射到这个范围内的核心；--bind-to core只是要求绑定到核心，但没有指定映射策略，所以OpenMPI会用默认的松散映射逻辑，导致分配混乱。

解决方法一：用--map-by core:pe=1严格映射 #

直接调整启动命令，加上--map-by core:pe=1参数，强制OpenMPI把每个进程精准映射到核心池中的单个核心：

mpirun --cpu-set 15-20 --map-by core:pe=1 --bind-to core -n 6 python3 bla.py

各参数的作用：

• --cpu-set 15-20：限制进程只能使用15到20号核心
• --map-by core:pe=1：指定按核心维度映射，每个核心仅运行1个进程（pe=1即每个处理单元分配1个进程）
• --bind-to core：强制每个进程绑定到分配的核心上，防止进程在核心间漂移

运行这个命令后，再执行你的脚本，cpu_affinity应该会显示15、16、17、18、19、20各出现一次，不会有重复或遗漏。

解决方法二：用hwloc语法精准绑定核心集 #

如果你的OpenMPI版本支持hwloc（大部分新版本都支持），可以用更精确的hwloc绑定参数，直接锁定目标核心：

mpirun --bind-to hwloc:cpuset=15-20 --map-by core -n 6 python3 bla.py

这里--bind-to hwloc:cpuset=15-20直接通过hwloc库指定核心范围，配合--map-by core确保每个进程对应一个独立核心，绑定效果会更稳定。

验证绑定是否生效 #

你可以在命令后追加--report-bindings和--display-map参数，直观查看绑定结果：

mpirun --cpu-set 15-20 --map-by core:pe=1 --bind-to core -n 6 --report-bindings --display-map python3 bla.py

正常情况下：

• --display-map会清晰显示每个进程绑定到15-20中的单个核心
• --report-bindings会输出每个rank对应的核心绑定信息
• 你的脚本输出的cpu_affinity也会是6个核心各出现一次

另外你提到之前用--map-by core不能和--cpu-set一起用，这大概率是旧版本OpenMPI的兼容性问题。建议用mpirun --version检查版本，升级到3.x及以上版本会解决这类参数冲突问题。

备注：内容来源于stack exchange，提问作者Pratyush Das