查看电脑已安装的CUDA、cudnn版本号方法

最新推荐文章于 2025-03-27 08:58:08 发布

巢子

最新推荐文章于 2025-03-27 08:58:08 发布

阅读量5.8k

点赞数 8

分类专栏：深度学习环境文章标签：深度学习

本文链接：https://blog.csdn.net/m0_49878324/article/details/143731865

版权

深度学习环境专栏收录该内容

1 篇文章

订阅专栏

一，打开终端

对于windows系统
直接ctrl+r，输入cmd即可打开终端
对于Ubuntu系统，直接右键选择“在终端打开”，即可打开终端

二，查看电脑CUDA的配置（电脑需要带有GPU）

终端输入：

nvidia-smi

显示如下：

| NVIDIA-SMI 560.28.03              Driver Version: 560.28.03      CUDA Version: 12.6     |
|-----------------------------------------+------------------------+----------------------+
| 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  Quadro P2000                   Off |   00000000:65:00.0  On |                  N/A |
| 53%   46C    P0             18W /   75W |     913MiB /   5120MiB |      0%      Default |
|                                         |                        |                  N/A |
+-----------------------------------------+------------------------+----------------------+
                                                                                         
+-----------------------------------------------------------------------------------------+
| Processes:                                                                              |
|  GPU   GI   CI        PID   Type   Process name                              GPU Memory |
|        ID   ID                                                               Usage      |
|=========================================================================================|
|    0   N/A  N/A      1152      G   /usr/lib/xorg/Xorg                            321MiB |
|    0   N/A  N/A      1297      G   /usr/bin/gnome-shell                          249MiB |
|    0   N/A  N/A      3796      G   /usr/lib/firefox/firefox                      182MiB |
|    0   N/A  N/A      4532      G   ...erProcess --variations-seed-version        155MiB |
+-----------------------------------------------------------------------------------------+

可以看到CUDA配置版本为12.6，驱动版本为560.28.03 。所以，安装的CUDA版本不能高于12.6，可以低于12.6，一般可以兼容。

查看已经安装的CUDA版本

终端输入：

nvcc -V

显示如下：

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2022 NVIDIA Corporation
Built on Sun_Oct_23_22:10:41_PDT_2022
Cuda compilation tools, release 11.4, V11.4.315
Build cuda_11.4.r11.4/compiler.31964100_0

可以看到，安装的CUDA版本的是11.4，低于要求的12.6，一般情况也是没问题的。

查看cudnn版本

终端输入：

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

cat为命令，/usr/local/cuda是安装的cuda路径，根据您安装的路径更改即可。
显示如下：

define CUDNN_MAJOR 8
define CUDNN_MINOR 2
define CUDNN_PATCHLEVEL 4
--
define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

endif /* CUDNN_VERSION_H */

所以，安装的cudnn版本是8.2.4.