cuda编程学习：获取gpu的数量和设置使用哪个gpu

#include <stdio.h>               
#include <cuda_runtime.h>        

int main()                      
{
    int iDeviceCount = 0;        // 定义一个变量来存储系统中的 CUDA 设备数量

    // 调用 cudaGetDeviceCount 获取设备数量，检查是否成功获取，或设备数量是否为0
    if (cudaGetDeviceCount(&iDeviceCount) != cudaSuccess || iDeviceCount == 0)
    {
        // 如果检测不到 CUDA 设备或有错误发生，打印错误信息并返回1
        printf("No CUDA devices found: %s\n", cudaGetErrorString(cudaGetLastError()));
        return 1; 
    }
    else
    {
        
        printf("CUDA device count: %d\n", iDeviceCount);
    }

    int iDev = 0;                // 定义一个变量 iDev，用于指定要使用的设备ID，这里设为0

    // 尝试设置当前线程的活动设备为 iDev（第0号设备）
    if (cudaSetDevice(iDev) != cudaSuccess)
    {
        // 如果设置设备失败，打印错误信息
        printf("CUDA set device error: %s\n", cudaGetErrorString(cudaGetLastError()));
    }
    else
    {
        // 如果设置设备成功，打印当前设置的设备ID
        printf("CUDA set device: %d\n", iDev);
    }

    return 0;                  
}