Ubuntu16.04 安装配置Caffe GPU-灵析社区

番茄炒蛋不要蛋

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,成功!!

阅读量:513

点赞量:0

收藏量:0