Ubuntu 如何查看 cuda 及 cudnn 版本及出现的问题

1、查看 CUDA 版本:

cat /usr/local/cuda/version.txt

或者:

nvcc -V

2、查看 CUDNN 版本:

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

但有可能会不显示,原因是因为cudnn的版本问题,每个版本cudnn.h文件内容不同导致,如下图两个版本的cudnn:

(1)cudnn-10.0-linux-x64-v7.4.2.24
Ubuntu 如何查看 cuda 及 cudnn 版本及出现的问题_第1张图片
该版本只有这一个文件。

(2)cudnn-10.2-linux-x64-v8.0.4.30:

Ubuntu 如何查看 cuda 及 cudnn 版本及出现的问题_第2张图片
点开cudnn.h文件就可以发现区别,所以查看这个高版本的cudnn需要改为:

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

Ubuntu 如何查看 cuda 及 cudnn 版本及出现的问题_第3张图片
注意:我电脑装了两个cuda和两个cudnn,一定要注意cuda和cudnn版本对应。

问题解决!

你可能感兴趣的:(linux,ubuntu,cuda,cudnn,nvidia)