如何在程序中返回 mpiexec -n 参数？

Posted 2023-03-27

【中文标题】如何在程序中返回 mpiexec -n 参数？

【英文标题】：How to return mpiexec -n parameter in program?

【发布时间】：2021-11-07 08:47:23

【问题描述】：

有没有办法在 Python 中使用 sys 模块返回一个 mpiexec -n 参数？

例如，假设我们在 Git Bash 中运行： mpiexec -n 6 python cuda.py

我们如何在程序中返回 -n 之后的数字？ os.environ 或某种系统模块（或单击）是否有助于实现这一目标？

【参考方案1】：

from mpi4py import MPI
comm = MPI.COMM_WORLD
nprocs = comm.Get_size()

您的 MPI 实现也有可能使环境变量可用，但这取决于实现。