cuda怎么看安装成功
时间: 2023-07-04 15:10:25 浏览: 163
在 Windows 操作系统下,你可以通过以下步骤来验证 CUDA 是否成功安装:
1. 打开命令提示符(Win+R,输入cmd,回车)。
2. 输入 `nvcc -V` 命令,查看 CUDA 版本信息。
如果你看到了类似于以下内容的输出,那么说明 CUDA 已经成功安装:
```
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2021 NVIDIA Corporation
Built on Sun_May__2_19:15:17_Pacific_Daylight_Time_2021
Cuda compilation tools, release 11.3, V11.3.109
Build cuda_11.3.r11.3/compiler.29920130_0
```
此外,你还可以编写一个简单的 CUDA 程序进行验证。例如,以下代码可以对两个向量进行加法:
```cuda
#include <stdio.h>
__global__ void add(int *a, int *b, int *c) {
int tid = blockIdx.x; // 获取当前线程的 ID
if (tid < 100)
c[tid] = a[tid] + b[tid];
}
int main() {
int a[100], b[100], c[100];
int *dev_a, *dev_b, *dev_c;
// 在 GPU 上分配内存
cudaMalloc((void **)&dev_a, 100 * sizeof(int));
cudaMalloc((void **)&dev_b, 100 * sizeof(int));
cudaMalloc((void **)&dev_c, 100 * sizeof(int));
// 初始化 a 和 b 数组
for (int i = 0; i < 100; ++i) {
a[i] = i;
b[i] = i * i;
}
// 将 a 和 b 数组拷贝到 GPU 上
cudaMemcpy(dev_a, a, 100 * sizeof(int), cudaMemcpyHostToDevice);
cudaMemcpy(dev_b, b, 100 * sizeof(int), cudaMemcpyHostToDevice);
// 在 GPU 上启动核函数
add<<<100, 1>>>(dev_a, dev_b, dev_c);
// 将结果拷贝回 CPU 上
cudaMemcpy(c, dev_c, 100 * sizeof(int), cudaMemcpyDeviceToHost);
// 打印结果
for (int i = 0; i < 100; ++i) {
printf("%d + %d = %d\n", a[i], b[i], c[i]);
}
// 释放 GPU 内存
cudaFree(dev_a);
cudaFree(dev_b);
cudaFree(dev_c);
return 0;
}
```
如果你能够成功编译并运行这个程序,并且得到正确的结果,那么说明 CUDA 已经成功安装。
阅读全文