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的路径进行调整