编写第1个CUDA程序

1. 环境准备

确保cuda驱动已经正确安装，执行以下命令查看：

nvidia-smi

有具体的显卡信息显示，表明显卡及其驱动都正常工作。若不正常工作可以参考教程安装驱动。

2. 源码准备

以下源码输出显卡支持的CUDA程序信息，cuda_status用于判断获取信息的状态，cuda_status为0表示获取成功。

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

using namespace std;

int main()
{
    int dev = 0;
    cudaDeviceProp dev_prop;
    cudaError_t cuda_status;
    cuda_status = cudaGetDeviceProperties(&dev_prop, dev);
    cout << "cuda_status:" << cuda_status << endl;
    cout << "使用GPU device " << dev << ": " << dev_prop.name << endl;
    cout << "SM的数量：" << dev_prop.multiProcessorCount << endl;
    cout << "每个线程块的共享内存大小：" << dev_prop.sharedMemPerBlock / 1024.0 << " KB" << endl;
    cout << "每个线程块的最大线程数：" << dev_prop.maxThreadsPerBlock << endl;
    cout << "每个SM的最大线程数：" << dev_prop.maxThreadsPerMultiProcessor << endl;
    cout << "每个SM的最大线程束数：" << dev_prop.maxThreadsPerMultiProcessor / 32 << endl;

    return 0;
}

3. 编译并运行

以上源码保存在文件GPU_info.cu中，编译命令如下：

nvcc  GPU_info.cu  -o gpu_info
./gpu_info

如下图所示，即为正确执行后的结果。