一、直接安装cudnn
二、查找合适版本安装
三、查看安装好的cudnn
安装之前先确定CUDA已安装成功。请点击此处查看如何在linux环境下查看自己的CUDA及版本。
一、直接安装
可在终端输入命令:conda install cudnn
即可安装cudnn,conda会自动匹配合适的版本。

二、如果上述方法未安装,可通过自己找合适的版本来安装。
1、查看是否安装好CUDA
nvcc -V
2、查看对应版本的cudnn(注意:仅查看,不需要下载)
https://developer.nvidia.com/rdp/cudnn-archive

3、安装cudnn
在终端输入命令,conda install cudnn=8.2.1​
注:这里的8.2.1要根据第2步中所查询到的版本​号。
三、查看是否安装成功
可通过conda list查看是否安装

可看到,cudnn已经安装,版本为8.2.1。

物联沃分享整理
物联沃-IOTWORD物联网 » linux下安装cudnn

发表评论