Ubuntu查看cuDNN版本

1、先成功下载cuDNN(如果这个有问题的,请参考我的另一个博客,地址:tensorflow2.0及以上版本用GPU加速计算_English ONly的博客-CSDN博客)

2、将下载成功的cudnn后得到的cuda里面的信息放到不同的 /usr/local/cuda目录里。(其中红色为下载cuDNN得到的cuda文件,绿色为我们在系统里配置CUDA环境变量的CUDA)

sudo cp cuda/include/cudnn_version.h /usr/local/cuda/include        #这一行的指令不是必须的,这一行指令会便于我们检测cuDNN的版本

sudo cp cuda/include/cudnn.h /usr/local/cuda/include

sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64

sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn* 

3、指令检测cuDNN版本

原指令:cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2      ##若该指令无输出是很正常的,因为新的版本的cuDNN的版本文件不在cudnn.h文件里,而是cudnn_version.h,这也是我们在2、中需要第一行指令的原因。

新指令:cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

4、使用新指令后的结果

(注:如果新指令没有效果,请将2、中的第一行指令认真看一下,不理解的请留言) 

最终得到我的cuDNN版本是8.2.1

你可能感兴趣的:(ubuntu,linux,cudnn,tensorflow,pytorch)