ubuntu查看cuda/cudnn版本

ubuntu查看cuda版本的命令：

nvcc -V
           

查看cudnn版本的命令（老版本）：

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

查看cudnn版本的命令（新版本）：

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

注意：可能cudnn不是装在/usr目录下的，需要根据自己的cudnn安装路径对cudnn.h或cudnn_version.h的路径进行调整