如何在 CUDA 中以编程方式获取卡规格

Posted 2023-04-15

【中文标题】如何在 CUDA 中以编程方式获取卡规格

【英文标题】：How to get card specs programmatically in CUDA

【发布时间】：2011-08-07 00:54:11

【问题描述】：

我刚刚开始使用 CUDA。有没有办法以编程方式获取卡规格？

【参考方案1】：

您可以使用cudaGetDeviceCount 和cudaGetDeviceProperties API。

void DisplayHeader()

    const int kb = 1024;
    const int mb = kb * kb;
    wcout << "NBody.GPU" << endl << "=========" << endl << endl;

    wcout << "CUDA version:   v" << CUDART_VERSION << endl;    
    wcout << "Thrust version: v" << THRUST_MAJOR_VERSION << "." << THRUST_MINOR_VERSION << endl << endl; 

    int devCount;
    cudaGetDeviceCount(&devCount);
    wcout << "CUDA Devices: " << endl << endl;

    for(int i = 0; i < devCount; ++i)
    
        cudaDeviceProp props;
        cudaGetDeviceProperties(&props, i);
        wcout << i << ": " << props.name << ": " << props.major << "." << props.minor << endl;
        wcout << "  Global memory:   " << props.totalGlobalMem / mb << "mb" << endl;
        wcout << "  Shared memory:   " << props.sharedMemPerBlock / kb << "kb" << endl;
        wcout << "  Constant memory: " << props.totalConstMem / kb << "kb" << endl;
        wcout << "  Block registers: " << props.regsPerBlock << endl << endl;

        wcout << "  Warp size:         " << props.warpSize << endl;
        wcout << "  Threads per block: " << props.maxThreadsPerBlock << endl;
        wcout << "  Max block dimensions: [ " << props.maxThreadsDim[0] << ", " << props.maxThreadsDim[1]  << ", " << props.maxThreadsDim[2] << " ]" << endl;
        wcout << "  Max grid dimensions:  [ " << props.maxGridSize[0] << ", " << props.maxGridSize[1]  << ", " << props.maxGridSize[2] << " ]" << endl;
        wcout << endl;
    

如果您已经安装了GPU Computing SDK，请查看%NVSDKCOMPUTE_ROOT%\C\src 目录中的deviceQuery 项目。它展示了如何使用 CUDA Runtime API 调用来查询所有设备属性。

CUDA Programming guide 在第 3.2.3 节中有更多详细信息。