一,打开终端
-
对于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.