cuda - 具有 Cuda Thrust 的多个 GPU？

Question

如何将 Thrust 与多个 GPU 结合使用？
是否只是使用 cudaSetDevice(deviceId) 的问题
然后运行相关的 Thrust 代码？

Answer 1

使用 CUDA 4.0 或更高版本，cudaSetDevice(deviceId)其次是您的推力代码应该可以工作。

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