#include
int main(void)
{
// 检测GPU数量
int iDeviceCount = 0;
cudaError_t error = cudaGetDeviceCount(&iDeviceCount);
if (error != cudaSuccess || iDeviceCount == 0)
{
printf("No CUDA campatable GPU found!\n");
exit(-1);
}
else
{
printf("The count of GPUs is %d.\n", iDeviceCount);
}
// 设置GPU执行
int iDevice = 0;
error = cudaSetDevice(iDevice);
if (error != cudaSuccess)
{
printf("fail to set GPU %d for computing.\n", iDevice);
exit(-1);
}
else
{
printf("set GPU %d for computing.\n", iDevice);
}
return 0;
}