Ubuntu下安装caffe,配置Python3.5.2接口

1. 安装caffe依赖包   

$ sudo apt-get install git
$ sudo apt-get install libprotobuf-dev
$ sudo apt-get install libleveldb-dev
$ sudo apt-get install libsnappy-dev
$ sudo apt-get install libopencv-dev
$ sudo apt-get install libhdf5-serial-dev
$ sudo apt-get install protobuf-compiler
$ sudo apt-get install --no-install-recommends
$ sudo apt-get install libboost-all-dev
$ sudo apt-get install libatlas-base-dev
$ sudo apt-get install python-dev
$ sudo apt-get install libgflags-dev
$ sudo apt-get install libgoogle-glog-dev
$ sudo apt-get install liblmdb-dev

2. 下载caffe

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

3. 修改 配置文件

首先,在caffe根目录下复制一份配置文件:

$ cd caffe
$ cp Makefile.config.example Makefile.config

然后,修改文件内容:

①CPU;②PYTHON_LIBRARIES; ③PYTHON_INCLUDE; ④INCLUDE_DIRS;⑤LIBRARY_DIRS.

...
...
# CPU-only switch (uncomment to build without GPU support).
CPU_ONLY := 1
...
...
# Uncomment to use Python 3 (default is Python 2)
PYTHON_LIBRARIES := boost_python-py35 python3.5m
PYTHON_INCLUDE := /usr/include/python3.5m \
                /usr/lib/python3/dist-packages/numpy/core/include
...
...
# Whatever else you find you need goes here.
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
...
...

4. 添加python环境 变量

在主目录下输入命令:

$ vim ~/.bashrc

按 ‘a’ 进入编辑模式,在文档末尾添加下列语句:

# added by Rain Jeyin in 2018/7/26
export PYTHONPATH=/home/user/caffe/python:$PYTHONPATH

注意:user为自己的主机用户名

然后按 ‘ESC’ 退出编辑模式,输入 ':wq' 保存退出

 在主目录下输入如下命令,使上述操作生效。

$ source ~/.bashrc

5. 开始编译

在caffe目录下完成:

$ make all -j4
$ make test -j4
$ make runtest -j4
$ make pycaffe

6. 验证

进入python, 导入caffe 模块

$ python
>>import caffe

7.  问题解决

报错了:ImportError: No module named 'google'

安装对应模块:

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值