(1) 去官网下载相应的版本,由于电脑以前安装的是 CUDA8.0 ,找跟 CUDA 版本兼容的 cuDNN 下载便可,我选择的是 cuDNN v7.1.4 Runtime Library for Ubuntu16.04(Deb) 和 cuDNN v7.1.4 Developer Library for Ubuntu 16.04 (Deb) html
注意这里下载安装包時须要账号登录才能够。linux
(2) 根据教程安装,主要步骤摘录以下ui
(3) 不过到这里先别着急,还须要后续操做一下,能够参考博客最后一部分spa
这是才算安装完成。code
(4) 而后再编译仍是找不到 libcudnn.so.7.1.4 ,怎么办?htm
以前好像跟 CUDA 连体安装的是 libcudnn.so.5.1.10 blog
cv@cv : /usr/local/cuda-8.0/include$ locate libcudnn.so /usr/lib/x86_64-linux-gnu/libcudnn.so /usr/lib/x86_64-linux-gnu/libcudnn.so.7 /usr/lib/x86_64-linux-gnu/libcudnn.so.7.1.4 /usr/local/cuda-8.0/lib64/libcudnn.so /usr/local/cuda-8.0/lib64/libcudnn.so.5 /usr/local/cuda-8.0/lib64/libcudnn.so.5.1.10
cmake 提示报错表示找不到新的版本教程
-- Found CUDNN: /usr/local/cuda-8.0/include -- Found cuDNN: v5.1.10 (include: /usr/local/cuda-8.0/include, library: /usr/local/cuda-8.0/lib64/libcudnn.so) CMake Error at libtorch/share/cmake/Caffe2/public/cuda.cmake:159 (message): PyTorch requires cuDNN 7 and above.
指向两个地方,下面咱们直接对症下药get
第一个是将 libcudnn.so 连接向 /usr/lib/x86_64-linux-gnu/libcudnn.so.7.1.4 博客
第二个是将 /usr/local/cuda-8.0/include 里面的 cudnn.h 改掉由于它是对应 libcudnn.so.5.1.10 的,怎么改呢,简单粗暴
cv@cv : /usr/local/cuda-8.0/include $ sudo mv cudnn.h cudnn.h.bak cv@cv : /usr/local/cuda-8.0/include $ locate cudnn.h /usr/include/cudnn.h /usr/local/cuda-8.0/include/cudnn.h cv@cv : /usr/local/cuda-8.0/include $ sudo cp -fv /usr/include/cudnn.h /usr/local/cuda-8.0/include/
(6) 查看一下内容,上面的是 cudnn 版本5的头文件部分摘录,下面的是 cudnn 版本7的头文件部分摘录。
到此应该能够经过 CMake 编译了, Bingo !