1、前面已经安装完显卡驱动和cuda8.0
2、安装caffe:
参考:Ubuntu16.04 安装配置Caffe - 一生不可自决 - 博客园
3、下载cuDNN:参考:Ubuntu16.04 Caffe 安装步骤记录(超详尽)_yhao的博客-CSDN博客_caffe安装
下载地址:https://developer.nvidia.com/rdp/cudnn-archive#a-collapse51b
选择:
4、参考链接 https://blog.csdn.net/yhaolpz/article/details/71375762 文件中的进入cuDNN5.1解压之后的include目录,在命令行进行如下操作。
严格按照
否则会编译caffe会报如下错误:
5、 make runtest过程中报错:意思是编译的过程中缺少libcudart.so.8.0文件
参考: https://blog.csdn.net/weixin_43439673/article/details/84198164
https://blog.csdn.net/w_q_q2017/article/details/83781976
可以解决以上问题:
具体过程如下:
然后再次make runtest,报错缺少libcudnn.so.5文件
同样:做如下操作
然后就可以再次make runtest了。
6、编译pycaffe
参考:https://www.cnblogs.com/go-better/p/7161006.html 中的编译pycaffe就可以了
按照这篇博客的步骤逐步操作即可:
唯一变的是.bashrc文件:
7、打开Python,成功!!
阅读量:508
点赞量:0
收藏量:0