查看cudnn和cuda版本

命令：
whereis cudnn.h
CUDNN_H_PATH=$(whereis cudnn.h)
cat ${CUDNN_H_PATH} | grep CUDNN_MAJOR -A 2

结果：
#define CUDNN_MAJOR 7
#define CUDNN_MINOR 5
#define CUDNN_PATCHLEVEL 0
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

你的cudnn版本是7.5.0

 

关于cuda：

如果使用pytorch，可以使用如下语句查询是否可用：

import torch

print(torch.__version__)  # 查看torch当前版本号

print(torch.version.cuda)  # 编译当前版本的torch使用的cuda版本号

print(torch.cuda.is_available())  # 查看当前cuda是否可用于当前版本的Torch，如果输出True，则表示可用

print(torch.backends.cudnn.version())

如果尚未安装torch，可以查看pytorch各版本与CUDA各版本的对应关系： https://pytorch.org/get-started/previous-versions/