我正在尝试跨两个gpus运行一个pycuda程序。我已经阅读了Talonmies的一篇精彩文章,解释了你如何使用线程库,帖子也提到了这可以用mpi4py。
当我用pycuda运行mpi4py时,程序会给出错误: self.ctx = driver.Device(gpuid).max_context pycuda._driver.logicError:cuDeviceGet失败:未初始化
也许这是因为我试图同时将两个gpu设备初始化。有没有人有一个非常简短的例子说明我们如何让2 gpus与mpi4py一起工作?