如何在程序中 return mpiexec -n 参数?

How to return mpiexec -n parameter in program?

有没有办法使用Python中的sys模块来return一个mpiexec -n参数?

例如,假设我们 运行 在 Git Bash: mpiexec -n 6 python cuda.py

我们如何return程序中-n之后的数字? os.environ 或某种系统模块(或单击)是否有助于实现此目的?

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

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