安装完cudnn后,使用以下命令查看cudnn版本没有反应
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
原因为NVIDIA把版本信息换到cudnn_version.h中了,因此使用以下命令查看cudnn版本信息。
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
若以上命令无反应。则进入到解压的cudnn文件夹中,复制cudnn_version.h到cuda文件夹中即可
sudo cp include/cudnn_version.h /usr/local/cuda/include/
再次执行查看版本命令,成功。