linux下安装cudnn

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

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

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

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

你可能感兴趣的:(linux,python,运维)