码迷,mamicode.com
首页 > 编程语言 > 详细

ubuntu 16.04 python3环境下安装仅cpu版本的caffe

时间:2020-06-11 13:31:39      阅读:86      评论:0      收藏:0      [点我收藏+]

标签:import   开启   sdn   packages   bashrc   使用   caff   include   aries   

1.安装依赖文件。

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

2.git clone https://github.com/BVLC/caffe.git 下载caffe源码。(个人认为用迅雷下载更快)

3.修改Makefile文件。

cd ~/caffe-master
cp Makefile.config.example Makefile.config
sudo gedit Makefile.config

删除 `#CPU_ONLY := 1` 的#,开启仅使用cpu。

以下两行带有hdf5的为新增加的内容,解决无法找到HD5的错误。

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/hdf5/serial

使用python3还需做如下修改,将PYTHON_INCLUDE python2的注释掉,取消python3的注释。
#python2
#PYTHON_INCLUDE := /usr/include/python2.7 \
/usr/lib/python2.7/dist-packages/numpy/core/include
#python3
PYTHON_LIBRARIES := boost_python3 python3.5m
PYTHON_INCLUDE := /usr/include/python3.5m \
/usr/lib/python3/dist-packages/numpy/core/include

4.解决`/usr/bin/ld: cannot find -lboost_python3` 错误
sudo ln -s /usr/lib/x86_64-linux-gnu/libboost_python-py35.so /usr/lib/x86_64-linux-gnu/libboost_python3.so
编译caffe,make默认单核运算,-j2表示双核,可加快速度,根据自己实际情况修改。
make all -j2
make test -j2
make runtest -j2

5.安装caffe python3接口。
sudo apt install python3-pip #安装pip工具。
sudo apt install python3-numpy==1.13.3  #解决后面编译无法找到numpy错误。
sudo apt-get install gfortran #安装依赖文件
sudo pip3 install matplotlib#解决后面import caffe报错

cd ~/caffe-master
pip3 install -r python/requirements.txt

6.编译
make pycaffe -j2

7.打开python3终端测试:

cd ~/caffe-master/python/
python3
import caffe #不报错则安装成功.

报错:ImportError: No module named ‘google‘

pip3 install google-cloud

pip3 install google-cloud-vision


8增加环境变量
sudo gedit ~/.bashrc
添加一行 export PYTHONPATH=~/caffe-master/python:$PYTHONPATH

 

参考:https://blog.csdn.net/weixin_38125866/article/details/81951548

ubuntu 16.04 python3环境下安装仅cpu版本的caffe

标签:import   开启   sdn   packages   bashrc   使用   caff   include   aries   

原文地址:https://www.cnblogs.com/jack123/p/13092361.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!