具有Cuda Thrust的多个GPU？

Question

如何在多个GPU中使用Thrust？ 是否只是使用cudaSetDevice（deviceId）的问题 然后运行相关的Thrust代码？

Answer 1

使用CUDA 4.0或更高版本，cudaSetDevice(deviceId)后跟您的推力代码应该有效。

请记住，您需要在每个设备上创建和操作单独的向量（除非您拥有支持对等内存访问的设备，并且PCI-express带宽足以满足您的任务）。