CUDA 装备操作实例

我的电脑上只有一个GPU，下面的例子就是先获取GPU的装备数量，然后设置GPU实行时使用的装备。

1. #include "cuda_runtime.h"
3. #include  <stdio.h>
5. int main()
6. {
7.         int nDeviceCount = 0;
8.         cudaError_t error = cudaGetDeviceCount(&nDeviceCount);
10.         if (error != cudaSuccess || nDeviceCount == 0)
11.         {
12.                 return -1;
13.         }
14.         printf("The count of GPUs is %d.\n", nDeviceCount);
16.         // 设置执行
17.         int iDevc = 0;
18.         error = cudaSetDevice(iDevc);
19.         if (error != cudaSuccess)
20.         {
21.                 return -1;
22.         }
24.         printf("Set GPU 0 for computing.\n");
26.         getchar();
28.         return 0;
29. }

复制代码

结果：

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。