要查看您的笔记本电脑上的GPU的CUDA Compute Capability,可以使用NVIDIA的命令行工具 nvidia-smi
或 deviceQuery
。这些工具通常在安装NVIDIA驱动程序和CUDA Toolkit后自动安装在您的系统中。
下面是如何使用这些工具来查看CUDA Compute Capability的步骤:
使用 nvidia-smi 命令:
- 打开终端或命令提示符。
- 输入以下命令并按回车键执行:
nvidia-smi
- 您将看到与您的GPU相关的信息,包括CUDA Compute Capability。查找包含 "Compute Capability" 的行,通常显示为 "C",后面跟着Compute Capability的版本号。例如:
| 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 Tesla K80 Off | 00000000:00:1E.0 Off | 0 |
| N/A 32C P0 59W / 149W | 0MiB / 11441MiB | 0% Default |
| | | N/A |
+-------------------------------+----------------------+----------------------+
| 1 Tesla K80 Off | 00000000:00:1F.0 Off | 0 |
| N/A 32C P0 56W / 149W | 0MiB / 11441MiB | 0% Default |
| | | N/A |
+-------------------------------+----------------------+----------------------+
Compute Capability: 3.7
在这个示例中,Compute Capability 是 3.7。
使用 deviceQuery 命令:
-
打开终端或命令提示符。
-
在终端中导航到 CUDA Toolkit 的
samples
目录。通常,这个目录位于您的CUDA Toolkit安装路径下,例如:
cd /usr/local/cuda/samples/1_Utilities/deviceQuery
3. 编译 deviceQuery
程序:
make
4. 运行 deviceQuery
:
./deviceQuery
5. 查找 "CUDA Capability Major/Minor version number" 行,这一行将显示Compute Capability 的版本号。例如:
CUDA Capability Major/Minor version number: 7.0
在这个示例中,Compute Capability 是 7.0
通过这些步骤,您可以轻松地确定您的GPU的CUDA Compute Capability。这对于选择与之兼容的深度学习框架和库非常重要。