python - 2 gpu で Pycuda を Mpi4py と連携させる

Question

2 つの gpu で pycuda プログラムを実行しようとしています。Talonmies によるスレッド ライブラリを使用した方法を説明したすばらしい投稿を読みました。この投稿では、mpi4py でこれが可能であると述べられています。

pycuda で mpi4py を実行すると、プログラムで次のエラーが表示されます:

おそらくこれは、2 つの GPU デバイスを同時に初期化しようとしたことが原因です。2 つの gpu を mpi4py で動作させる方法の非常に短い例はありますか?

Answer 1

この質問に出くわした人のために、これは実際の mpi4py+pycuda の例です。