安装cuda toolkit 和 cudnn

安装cuda toolkit

https://developer.nvidia.com/cuda-toolkit-archive

选择自己对应的版本

安装cudnn

https://developer.nvidia.com/rdp/cudnn-archive

选择自己对应的版本

装完后依次执行：

如果是deb文件

sudo dpkg -i cudnn-xxx

如果是tar文件

tar -xvf cudnn-xxx

sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include 

sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64 

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

检查

cat /usr/local/cuda-xxx/include/cudnn_version.h | grep CUDNN_MAJOR -A 2