选择设备

system · 2013 年10 月 22 日 02:52

大家好，我现在有两块GPU一个是gtx560 一个是gtx680，我现在程序已经写好了，我想问一下如果我想比较两个显卡的执行速度，是不是只需要在main函数开始处修改cudaSetDevice(0) 或者 cudaSetDevice(1)就可以了呢？

system · 2013 年10 月 22 日 02:55

LZ您好：

您可以通过cudaSetDevice()来选择您的代码在您机器上的哪个GPU上执行。

祝您好运~

system · 2013 年10 月 22 日 05:31

是不是只需要配置一次呢？配饰一次后面的代码就都在选定的GPU上执行吗？

system · 2013 年10 月 22 日 05:32

楼主您好，是这样的。您的理解是正确的。

同时，不仅仅是kernel都在该GPU上执行，cudaMalloc分配之类的，也将在该卡上执行。

感谢来访。

system · 2013 年10 月 22 日 05:38

好的谢谢版主