📅  最后修改于: 2023-12-03 14:52:09.607000             🧑  作者: Mango
如果您是一位使用 NVIDIA GPU 的程序员,那么您可能需要安装 CUDA 来加速您的计算任务。在安装 CUDA 之后,您可以使用以下几种方法来判断您是否已成功安装:
在终端窗口中运行以下命令:
nvidia-smi
如果安装了 CUDA 和 NVIDIA 驱动程序,您应该会看到一个包含以下信息的表格:
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 410.93 Driver Version: 410.93 CUDA Version: 10.0 |
|-------------------------------+----------------------+----------------------+
| GPU Name TCC/WDDM | Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
|===============================+======================+======================|
| 0 GeForce GTX 970M WDDM | 00000001:01:00.0 Off | N/A |
| N/A 48C P0 26W / N/A | 226MiB / 3072MiB | 0% Default |
+-------------------------------+----------------------+----------------------+
您可以在表格中找到 NVIDIA 驱动程序和 CUDA 的版本信息。如果没有显示该表格,那么您可能需要检查您的 NVIDIA 驱动程序和 CUDA 是否正确安装。
在终端窗口中运行以下命令:
nvcc --version
如果安装了 CUDA,您应该会看到类似以下信息的输出:
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2018 NVIDIA Corporation
Built on Sat_Aug_25_21:08:01_CDT_2018
Cuda compilation tools, release 10.0, V10.0.130
在终端窗口中运行以下命令:
echo $LD_LIBRARY_PATH
如果安装了 CUDA,您应该会看到类似以下信息的输出:
/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64
您可以通过检查 LD_LIBRARY_PATH
变量来确定 CUDA 是否正确安装并设置了路径。
在终端窗口中运行以下命令:
cd /usr/local/cuda-10.0/samples/0_Simple/vectorAdd/
make
./vectorAdd
如果安装了 CUDA,您应该会看到类似以下信息的输出:
...
Test PASSED
Done
这表明 CUDA 程序可以成功执行。
以上是几种判断您是否已成功安装 CUDA 的方法。如果您发现 CUDA 无法正常工作,请检查您的 NVIDIA 驱动程序、CUDA 安装和环境变量设置。