2 gpusのMpi4pyで作業するピクチャを取得する

Question

2つのgpusでpycudaプログラムを実行しようとしています。私はTalonmiesの素晴らしい投稿を読んできました。あなたがスレッドライブラリでどのようにそれを行うのかを説明しています。投稿はmpi4pyでも可能です。

私はpycudaでmpi4pyを実行すると、プログラムはエラーを与える： self.ctx = driver.Device（gpuidは） pycuda._driver.logicErrorを.max_context：cuDeviceGetに失敗しました：

を初期化していないおそらくこれがためです私の試みは、gpuデバイスの2つを同時に初期化することです。誰かがmpi4pyを使って2つのgpusを動作させる方法の非常に短い例がありますか？

Answer 1

hereはこの問題の可能性が高い人のために、mpi4py + pycudaの実例です。