cudnn安装
2019-03-10 本文已影响0人
王金松
https://developer.nvidia.com/rdp/cudnn-download
https://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html#installlinux-tar
下载
之前下载和安装的驱动和cuda版本如下
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 418.43 Driver Version: 418.43 CUDA Version: 10.1 |
|-------------------------------+----------------------+----------------------+
所以我们只需要安装对应的cudnn即可
重命名格式
cp cudnn-10.1-linux-x64-v7.5.0.56.solitairetheme8 cudnn-10.1-linux-x64-v7.5.0.56.tgz
解压
tar -xzvf cudnn-10.1-linux-x64-v7.5.0.56.tgz
#解压之后会生成cuda文件件
拷贝文件
$ 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*
