本人所用的系统为linux Ubuntu18.04，安装docker或其他框架经常要查询cuda和cudnn版本，总结如下：

查看cuda版本

方法1、

nvcc -V 
或者
nvcc --version

如果nvcc没有安装，用方法2

方法2、

cat /usr/local/cuda/version.txt

但是可能报错，提示如下：

cat: /usr/local/cuda/version.txt: No such file or directory

进入此目录，发现version.txt 换成了 version.json,可适用以下命令：

cat /usr/local/cuda/version.json

查看cudnn版本

方法1、

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

如果没反应，查询不到，可试下方法2

方法2、

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

如果以上方法不起作用，试试torch的api；
首先确定已经安装了torch，否则需要安装一下：pip install torch -i https://pypi.tuna.tsinghua.edu.cn/simple
进入python环境，输入以下代码：

import torch

#查看torch的版本
print(torch.__version__)

#查看cuda的版本
print(torch.version.cuda)

#查看cudnn的版本
print(torch.backends.cudnn.version())

有问题欢迎随时联系~

彩蛋 

更多干货内容请关注公众号SLearningAI，欢迎CSDN与公众号同步交流^_^