如何在程序中 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 实现也可能使环境变量可用,但这取决于实现。
有没有办法使用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 实现也可能使环境变量可用,但这取决于实现。