cudnn8.1.0安装

前提已安装cuda11.2或其他版本

官网下载：cuDNN Library for Linux (x86_64)

链接: nvidia官网

安装

tar -xzvf cudnn-11.2-linux-x64-v8.1.0.77.tgz  #解压
sudo cp cuda/include/*  /usr/local/cuda-11.2/include/ 
sudo cp cuda/lib64/libcudnn*  /usr/local/cuda-11.2/lib64/
sudo chmod a+r /usr/local/cuda-11.2/include/*
sudo chmod a+r /usr/local/cuda-11.2/lib64/libcudnn*

输入以下命令查看cudnn是否安装成功
cat /usr/local/cuda-11.2/include/cudnn_version.h | grep CUDNN_MAJOR -A 2