【linux】验证cuda是否安装成功

【linux】验证cuda是否安装成功

【创作不易，求点赞+关注+收藏】
确认 CUDA 是否完全安装并正确配置，可以运行一些简单的 CUDA 示例代码。以下是一个基本的 CUDA 示例代码：

1、创建cuda源文件

在终端中运行以下命令，使用 nano 文本编辑器创建一个新的文件 vector_add.cu：

nano vector_add.cu

将以下代码复制并粘贴到文件中：

#include <stdio.h>

__global__ void vector_add(float *a, float *b, float *c, int n) {
    int id = blockIdx.x * blockDim.x + threadIdx.x;
    if (id < n) {
        c[id] = a[id] + b[id];
    }
}

int main() {
    int n = 1000;
    float *a, *b, *c;
    float *d_a, *d_b, *d_c;
    
    a = (float*)malloc(n * sizeof(float));
    b = (float*)malloc(n * sizeof(float));
    c = (float*)ma