CUDA 检测GPU数量的C++代码

#include <stdio.h>


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;
}