https://developer.nvidia.com/cudnn
选择合适的版本下载(cuDNN Library for Linux),
然后解压,并进入到相应目录,运行以下命令:
sudo cp cuda/include/cud* /usr/local/cuda-10.1/include
sudo cp cuda/lib64/lib* /usr/local/cuda-10.1/lib64
sudo chmod a+r /usr/local/cuda-10.1/include/cudnn.h
sudo chmod a+r /usr/local/cuda-10.1/lib64/libcudnn*
以配置cuDNN环境。
通过测试是否安装成功。
cat /usr/local/cuda-10.1/include/cudnn_version.h | grep CUDNN_MAJOR -A 2