cuda的安装可以参考我的前一篇文章:
https://blog.csdn.net/weixin_40671425/article/details/82942375
去官网下载cudnn文件,
https://developer.nvidia.com/rdp/cudnn-download
需要登录账号
选择对应cuda版本的cudnn
下载是一个solitairetheme8文件,可以上传到服务器上
分别执行以下步骤即可:
cp cudnn-9.0-linux-x64-v7.3.1.20.solitairetheme8 cudnn-9.0-linux-x64-v7.3.1.20.tgz
tzr -zxvf cudnn-9.0-linux-x64-v7.3.1.20.tgz
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*