安装cudnn

1. 到官网下载cudnn（需要先注册）

wget https://..............

2. 解压下载的软件包

tar -xzvf cudnn-10.1-linux-x64-v7.5.0.56.tgz

3. 将解压的文件拷贝到相应的文件夹

sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*