Linux环境cudnn安装

Posted 2022-10-25 AI 菌

1、下载cudnn下载

• 首先去nvidia官网下载对应版本的cudnn
• 下载完成后，进行解压：tar zxvf cudnn-10.2-linux-x64-v7.6.5.32.tgz

2、删除已经安装的cudnn版本

• 没安装过cudnn的可以忽略此步骤
• 删除已经安装的cudnn版本

cd /usr/local/cuda/include/ 
rm -r cudnn.h
cd /usr/local/cuda/lib64/ 
rm -r libcudnn*

3、复制cudnn文件到指定路径

• 复制文件到指定路径

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/  
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/  

• 修改访问权限

sudo chmod 777  /usr/local/cuda/include/cudnn.h  
sudo chmod 777 /usr/local/cuda/lib64/libcudnn*

4、检查安装

• 输入命令：cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2