【深度学习环境配置踩坑记】二. ubuntu16.04+anaconda3安装caffe

caffe需要的前期准备包括anaconda3+显卡驱动+cuda+cudnn请参见这两篇博客:

https://blog.csdn.net/zl3090/article/details/89297031

https://blog.csdn.net/zl3090/article/details/89304484

 

接下来进入正题,本博文主要参考:

https://blog.csdn.net/yhaolpz/article/details/71375762

https://blog.csdn.net/w113691/article/details/77942408

 

1. 首先在你要安装的路径下 clone :

git clone https://github.com/BVLC/caffe.git

进入 caffe ,将 Makefile.config.example 文件复制一份并更名为 Makefile.config ,也可以在 caffe 目录下直接调用以下命令完成复制操作 :

sudo cp Makefile.config.example Makefile.config

复制一份的原因是编译 caffe 时需要的是 Makefile.config 文件,而Makefile.config.example 只是caffe 给出的配置文件例子,不能用来编译 caffe。

2. 然后修改 Makefile.config 文件,在 caffe 目录下打开该文件:

sudo gedit Makefile.config

(1)应用 cudnn

将
#USE_CUDNN := 1
修改成: 
USE_CUDNN := 1

(2)应用 opencv 版本

USE_OPENCV := 1

#OPENCV_VERSION := 3 这句话不改,否则会报错

(3)使用 python 接口

将
#WITH_PYTHON_LAYER := 1
修改为
WITH_PYTHON_LAYER := 1

(4)修改 python 路径

INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib
修改为:
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu /usr/lib/x86_64-linux-gnu/hdf5/serial 

3. 修改Makefile文件

(1)

将: 
NVCCFLAGS +=-ccbin=$(CXX) -Xcompiler-fPIC $(COMMON_FLAGS) 
替换为: 
NVCCFLAGS += -D_FORCE_INLINES -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS)
!

(2)

将: 
LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_hl hdf5 
改为: 
LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_serial_hl hdf5_serial

(3)

查找“Derive include and lib directories”一节,
修改“LIBRARIES +=”的最后一行(LIBRARIES +=opencv_imgcodecs ),增加opencv_imgcodecs

opencv_core opencv_highgui opencv_imgproc opencv_imgcodecs

此时应该变为LIBRARIES += opencv_imgcodecs  opencv_core opencv_highgui opencv_imgproc opencv_imgcodecs
!

4.修改 /usr/local/cuda/include/host_config.h 文件 :

将 
#error-- unsupported GNU version! gcc versions later than 4.9 are not supported! 
改为 
//#error-- unsupported GNU version! gcc versions later than 4.9 are not supported!

5. make

make all -j8

这是如果之前的配置或安装出错,那么编译就会出现各种各样的问题,所以前面的步骤一定要细心。

编译成功后可运行测试:

sudo make runtest -j8

如果显示结果为上图所示,则表示 caffe 已经成功安装。

6. 安装 pycaffe notebook 接口环境

在上一步成功安装 caffe 之后,就可以通过 caffe 去做训练数据集或者预测各种相关的事了,只不过需要在命令行下通过 caffe 命令进行操作,而这一步 pycaffe 的安装以及 notebook 环境配置只是为了更方便的去使用 caffe ,实际上大多数都是通过 python 来操作 caffe 的,而 notebook 使用浏览器作为界面,可以更方便的编写和执行 python 代码。

首先编译 pycaffe :

cd caffe

sudo make pycaffe -j8

编译 pycaffe 成功后,验证一下是否可以在 python 中导入 caffe 包,首先进入 python 环境

>>> import caffe

若不报错则表示 caffe 的 python 接口已正确编译

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值