一、直接安装cudnn
二、查找合适版本安装
三、查看安装好的cudnn
安装之前先确定CUDA已安装成功。请点击此处查看如何在linux环境下查看自己的CUDA及版本。
一、直接安装
可在终端输入命令:conda install cudnn
即可安装cudnn,conda会自动匹配合适的版本。
二、如果上述方法未安装,可通过自己找合适的版本来安装。
1、查看是否安装好CUDA
nvcc -V
2、查看对应版本的cudnn(注意:仅查看,不需要下载)
https://developer.nvidia.com/rdp/cudnn-archive
安装cudnn
在终端输入命令,conda install cudnn=8.2.1
注:这里的8.2.1要根据第2步中所查询到的版本号。
三、查看是否安装成功
可通过conda list查看是否安装
可看到,cudnn已经安装,版本为8.2.1。
来源地址:https://blog.csdn.net/zhchjcq1/article/details/130551652