CUDA查询和选取设备信息

CUDA C中的cudaGetDeviceProperties函数可以很方便的获取到设备的信息。

#include "cuda_runtime.h"
#include "device_launch_parameters.h"
#include <iostream>

using namespace std;

int main()
{
    cudaDeviceProp deviceProp;
    int deviceCount;
    cudaError_t cudaError;
    cudaError = cudaGetDeviceCount(&deviceCount);
    for (int i = 0; i < deviceCount; i++)
    {
        cudaError = cudaGetDeviceProperties(&deviceProp, i);

        cout << "设备 " << i + 1 << " 的主要属性： " << endl;
        cout << "设备显卡型号： " << deviceProp.name << endl;
        cout << "设备全局内存总量（以MB为单位）： " << deviceProp.totalGlobalMem / 1024 / 1024 << endl;
        cout << "设备上一个线程块（Block）中可用的最大共享内存（以KB为单位）： " << deviceProp.sharedMemPerBlock / 1024 << endl;
        cout << "设备上一个线程块（Block）种可用的32位寄存器数量： " << deviceProp.regsPerBlock << endl;
        cout << "设备上一个线程块（Block）可包含的最大线程数量： " << deviceProp.maxThreadsPerBlock << endl;
        cout << "设备的计算功能集（Compute Capability）的版本号： " << deviceProp.major << "." << deviceProp.minor << endl;
        cout << "设备上多处理器的数量： " << deviceProp.multiProcessorCount << endl;
    }
    getchar();
    return 0;
    }

获取设备，设置上下文的设备

#include "cuda_runtime.h"
#include "device_launch_parameters.h"
#include <iostream>
#include <cstring>
#include<cstdio>
using namespace std;

int main(){
    //定义需要的设备的属性
    cudaDeviceProp devicePropDefined;
    memset(&devicePropDefined,0,sizeof(cudaDeviceProp));
    devicePropDefined.major=5;  //设备的计算功能集（Compute Capability）的版本号
    devicePropDefined.minor=2;

    int devicedChoosed; //选中的设备的ID
    cudaError_t cudaError;
    cudaGetDevice(&devicedChoosed); //获取当前设备的ID

    cout << "choose device is " << devicedChoosed << endl;

    cudaChooseDevice(&devicedChoosed, &devicePropDefined);  //查找符合要求的设备ID
    cout << "choose device in request is 0 " << devicedChoosed << endl;
    cudaError = cudaSetDevice(devicedChoosed); //设置选中的设备为下文的运行设备

    if (cudaError == cudaSuccess)
        cout << "Success!" << endl;
    else
        cout << "Error!" << endl;
    getchar();
    return 0;

}

输出结果：

choose device is 0
choose device in request is 0
Success!