📜  如何判断我是否安装了 cuda - Shell-Bash (1)

📅  最后修改于: 2023-12-03 14:52:09.607000             🧑  作者: Mango

如何判断我是否安装了 CUDA

如果您是一位使用 NVIDIA GPU 的程序员,那么您可能需要安装 CUDA 来加速您的计算任务。在安装 CUDA 之后,您可以使用以下几种方法来判断您是否已成功安装:

方法一:查看 NVIDIA 驱动程序

在终端窗口中运行以下命令:

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 是否正确安装。

方法二:查看 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 是否正确安装并设置了路径。

方法四:执行 CUDA 示例代码

在终端窗口中运行以下命令:

cd /usr/local/cuda-10.0/samples/0_Simple/vectorAdd/
make
./vectorAdd

如果安装了 CUDA,您应该会看到类似以下信息的输出:

...
Test PASSED
Done

这表明 CUDA 程序可以成功执行。

以上是几种判断您是否已成功安装 CUDA 的方法。如果您发现 CUDA 无法正常工作,请检查您的 NVIDIA 驱动程序、CUDA 安装和环境变量设置。