numbers - 为 openmpi 设置处理器数量

Question

目前我正在使用这样的东西来运行一个 c++ 程序：

mpirun -np 4 ./test

这很好用，但问题是我正在用 Python 包装 c++ 程序。所以我想在 Python 中设置处理器的数量并将其传递给 c++ 程序。

是否可以在不使用命令行的情况下将使用的处理器数量设置为 c++ 代码中的固定值？一个简单的例子是什么样子的？

Answer 1

通过 -np N ，您设置进程数，而不是处理器数。根据您在执行期间提供的标志或排名文件（如果您正在使用），处理器被进一步分配给每个进程。如果您专注于 Python 和 C++ 交互，我认为应该有一种方法可以调用 cmd 函数，例如 exec()。您可以轻松地在系统命令行上调用您想要执行的任何内容！