我的电脑上只有一个GPU,下面的例子就是先获取GPU的设备数量,然后设置GPU执行时使用的设备。
#include "cuda_runtime.h"
#include <stdio.h>
int main()
{
int nDeviceCount = 0;
cudaError_t error = cudaGetDeviceCount(&nDeviceCount);
if (error != cudaSuccess || nDeviceCount == 0)
{
return -1;
}
printf("The count of GPUs is %d.\n", nDeviceCount);
// 设置执行
int iDevc = 0;
error = cudaSetDevice(iDevc);
if (error != cudaSuccess)
{
return -1;
}
printf("Set GPU 0 for computing.\n");
getchar();
return 0;
}
结果: