让Pycuda在2 gpus上使用Mpi4py

时间:2011-07-15 20:00:06

标签: python cuda mpi gpu pycuda

我正在尝试跨两个gpus运行一个pycuda程序。我已经阅读了Talonmies的一篇精彩文章,解释了你如何使用线程库,帖子也提到了这可以用mpi4py。

当我用pycuda运行mpi4py时,程序会给出错误:   self.ctx = driver.Device(gpuid).max_context   pycuda._driver.logicError:cuDeviceGet失败:未初始化

也许这是因为我试图同时将两个gpu设备初始化。有没有人有一个非常简短的例子说明我们如何让2 gpus与mpi4py一起工作?

1 个答案:

答案 0 :(得分:1)

对于任何有机会解决这个问题的人来说,here是一个有用的mpi4py + pycuda示例。