检查是否安装了 CUDA

如果您在使用 NVIDIA GPU 进行深度学习模型训练或其他 CUDA 加速工作时遇到问题，您需要检查是否已正确安装 CUDA。以下是使用 Bash Shell 在 Linux 上检查 CUDA 安装的常见方法。

1. 检查 NVIDIA 驱动程序

首先，请检查您的 NVIDIA 显卡是否正确安装并且正在运行适当的 NVIDIA 驱动程序。您可以使用以下命令检查驱动程序是否正确安装：

nvidia-smi

如果您看到 NVIDIA 显卡的基本信息和当前正在使用的驱动程序版本，则说明驱动程序已正确安装并且正在运行：

| NVIDIA-SMI 465.27       Driver Version: 465.27       CUDA Version: 11.3     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                               |                      |               MIG M. |
|===============================+======================+======================+
|   0  GeForce RTX 3070    Off  | 00000000:01:00.0  On |                  N/A |
|  0%   49C    P8    25W / 220W |    514MiB /  7982MiB |      4%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+

如果您未看到此输出或看到的是不明或错误的消息，则可能需要重新安装驱动程序。

2. 检查 CUDA 环境变量

安装 CUDA 后，您还需要设置一些环境变量以确保您的系统正确配置为使用 CUDA。您可以使用以下命令检查您的终端是否已正确设置这些变量：

echo $PATH | grep /usr/local/cuda/bin

如果输出中包含 /usr/local/cuda/bin，则 CUDA 环境变量已正确配置：

/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/games:/usr/local/games:/snap/bin

如果您看到命令未返回任何输出，则需要手动添加以下行到您的 ~/.bashrc 或 ~/.bash_profile 文件中：

export PATH=/usr/local/cuda/bin:$PATH

执行以下命令以使更改生效：

source ~/.bashrc

3. 检查 CUDA 版本

如果您安装了多个版本的 CUDA，您需要检查您当前运行的是哪个版本。您可以使用以下命令检查 CUDA 版本：

nvcc --version

如果您看到 CUDA 版本号，则您已正确安装 CUDA：

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2021 NVIDIA Corporation
Built on Wed_Jul_14_19:58:21_PDT_2021
Cuda compilation tools, release 11.4, V11.4.100
Build cuda_11.4.r11.4/compiler.30188945_0

否则，您需要重新安装 CUDA。

结论

使用上述命令，您可以轻松地检查您的 CUDA 安装是否正确。如果您还遇到任何问题，请查看 NVIDIA 的文档或参考相关的论坛帖子。