讓Pycuda在2 gpus上使用Mpi4py

Question

我試圖在兩個gpus上運行pycuda程序。我已經閱讀了Talonmies的一篇很好的文章，解釋瞭如何使用線程庫來實現，帖子也提到這可以通過mpi4py來實現。

當我運行pycuda mpi4py，程序給出了錯誤： self.ctx = driver.Device（gpuid）.max_context pycuda._driver.logicError：cuDeviceGet失敗：未初始化

也許這是因爲我試圖同時使用兩個gpu設備。有沒有人有一個很簡單的例子，我們如何能得到2 gpus與mpi4py工作？

Answer 1

對於任何有此機會的人，here是一個可用的mpi4py + pycuda示例。