检查版本 GPU

在进行深度学习等需要大量计算的任务时，GPU 是一个非常重要的硬件设备。为了保证代码的正确性和效率，我们需要检查 GPU 的版本信息。本文将介绍在 Shell/Bash 环境下检查 GPU 版本信息的常用方法。

方式一：使用 nvidia-smi 命令

nvidia-smi 是 NVIDIA 提供的一个用于查看 GPU 状态的命令行工具，它可以以表格或 XML 格式输出 GPU 的状态信息，包括 GPU 的名称、驱动版本、当前 GPU 利用率等等。

运行以下命令可以查看当前系统中所有的 GPU 设备和它们的驱动版本：

nvidia-smi

运行结果可能长成这个样子：

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 415.27       Driver Version: 415.27       CUDA Version: 10.0     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  Tesla K80           Off  | 00000000:00:04.0 Off |                    0 |
| N/A   33C    P0    74W / 149W |      0MiB / 11441MiB |      0%      Default |
+-------------------------------+----------------------+----------------------+
|   1  Tesla K80           Off  | 00000000:00:05.0 Off |                    0 |
| N/A   42C    P0    73W / 149W |      0MiB / 11441MiB |      5%      Default |
+-------------------------------+----------------------+----------------------+
|   2  Tesla K80           Off  | 00000000:00:06.0 Off |                    0 |
| N/A   39C    P0    73W / 149W |      0MiB / 11441MiB |      5%      Default |
+-------------------------------+----------------------+----------------------+
|   3  Tesla K80           Off  | 00000000:00:07.0 Off |                    0 |
| N/A   34C    P0    72W / 149W |      0MiB / 11441MiB |      4%      Default |
+-------------------------------+----------------------+----------------------+

...

在这个示例中，我们可以看到当前系统中有 4 个 Tesla K80 GPU 设备，其驱动版本为 415.27。

方式二：查看 CUDA 版本

当我们在使用 NVIDIA GPU 进行深度学习等任务时，通常还需要检查 CUDA 工具包的版本。CUDA 可以加速运算，因此正确安装和使用 CUDA 是非常重要的。

可以通过以下命令检查 CUDA 版本：

nvcc --version

运行结果可能长成这个样子：

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

在这个示例中，我们可以看到当前系统中装有 CUDA 10.0 工具包。

结论

通过本文，您学会了在 Shell/Bash 环境下如何查看 GPU 和 CUDA 工具包的版本信息。这些知识对于深度学习等任务的正确性和效率是非常关键的。