python - Python スクリプトは mpirun または mpiexec によって起動されましたか?

Question

スタンドアロンで起動するか、mpi サポートを使用して起動できる Python スクリプトをコーディングしました。

python myscript.py

対

mpirun -np 2 python myscript.py

条件付き操作を実行するためにスクリプトが起動された方法をスクリプト内で知るにはどうすればよいですか?

Answer 1

Unix を使用している場合は、次の出力を分析できます。

import os 
print os.popen("ps -p %d -oargs=" % os.getpid()).read().strip()

getpid()andで遊んでgetppid()ください（親のために）。移植可能なソリューションには、 psutilなどの外部ライブラリが必要です。

import psutil, os
p = psutil.Process(os.getppid())
print p.name