(1) 去官网下载相应的版本,因为电脑之前安装的是 CUDA8. ,找跟 CUDA 版本兼容的 cuDNN 下载即可,我选择的是 cuDNN v7.1.4 Runtime Library for Ubuntu16.(Deb) 和  cuDNN v7.1.4 Developer Library for Ubuntu 16.04 (Deb)

注意这里下载安装包時需要帐号登陆才可以。

(2) 根据教程安装,主要步骤摘录如下

(3) 不过到这里先别着急,还需要后续操作一下,可以参考博客最后一部分

这是才算安装完成。

(4) 然后再编译还是找不到 libcudnn.so.7.1. ,怎么办?

之前好像跟 CUDA 连体安装的是 libcudnn.so.5.1.

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.

指向两个地方,下面我们直接对症下药

第一个是将 libcudnn.so 链接向 /usr/lib/x86_64-linux-gnu/libcudnn.so.7.1.

第二个是将 /usr/local/cuda-8.0/include 里面的 cudnn.h 改掉因为它是对应 libcudnn.so.5.1. 的,怎么改呢,简单粗暴

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 !

最新文章

  1. oracle--存储过程--bai
  2. PowerDesigner生成sql及HTML格式数据库文档
  3. 转 A Week with Mozilla's Rust
  4. 【poj2778-DNA Sequence】AC自动机+矩阵乘法
  5. win7下搭建opengles2.0编程环境
  6. 阿里云服务器(Win 2008 R2 Standard)安装MSSM 2008 R2之1033和2052问题
  7. 用c++语言编写函数 int index(char *s,char * t),返回字符串t在字符串s中出现的最左边的位置,如果s中没有与t匹配的子串,则返回-1。类似于索引的功能。
  8. DLX 舞蹈链 精确覆盖 与 重复覆盖
  9. SQL Server查看表结构及视图,适合开发者使用,简单易用
  10. BZOJ_4554_[Tjoi2016&Heoi2016]游戏_二分图匹配
  11. C++回顾day03---<多态>
  12. 阶段02JavaWeb基础day02&03JavaScript
  13. C#中读取xml文件指定节点
  14. MapReduce实例&YARN框架
  15. DNS之XX记录
  16. C++面向对象高级编程(八)模板
  17. codechef January Lunchtime 2017简要题解
  18. Ros学习——Cmakelists.txt文件解读
  19. linux非阻塞的socket EAGAIN的错误处理【转】
  20. iOS开发:iPhone6、6 plus适配

热门文章

  1. 【Android - 进阶】之Animation补间动画
  2. CentOS 7 ETCD集群配置大全
  3. 如何切换本地的GIT账号
  4. 【HUAWEI Mate30】抽奖啦!华为IoT新福利上线!
  5. MySql CPU彪高到百分之1000的排查思路
  6. Kafka 安装配置 及 简单实验记录
  7. Mysql的查询语句的使用
  8. luogu P1908 逆序对 |树状数组
  9. Java的值类型和引用类型
  10. Golang 在电商即时通讯服务建设中的实践