您的位置:首页 > 编程语言 > Python开发

ubuntu 16.04安装caffe(python2)

2017-04-18 16:21 573 查看
注:本文适合python2.7  opencv2.4.x, 之前学习其他DL框架时安装过CUDA的情况

一、安装必要的依赖库

sudo apt-get update

sudo apt-get install -y libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libboost-all-dev hdf5-tools libhdf5-serial-dev protobuf-compiler gfortran libjpeg62 libfreeimage-dev
libatlas-base-dev git python-dev python-pip libgoogle-glog-dev libbz2-dev libxml2-dev libxslt-dev libffi-dev libssl-dev libgflags-dev liblmdb-dev python-yaml


二、下载caffe并安装python依赖库

cd ~
git clone https://github.com/BVLC/caffe.git cd caffe
cat python/requirements.txt | xargs -L 1 sudo pip install


三、增加符号链接

sudo ln -s /usr/include/python2.7/ /usr/local/include/python2.7
sudo ln -s /usr/local/lib/python2.7/dist-packages/numpy/core/include/numpy/ /usr/local/include/python2.7/numpy


四、生成Makefile.config并修改

       生成文件:
cp Makefile.config.example Makefile.config


        打开Makefile.config文件:

       将标签为“INCLUDE_DIRS”和“LIBRARY_DIRS”修改如下:

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

特别提醒:将WITH_PYTHON_LAYER := 1 前面的#去掉,有些程序(比如fast-rcnn)会在proto里面写入type: "Python",去掉这行的注释才能正常运行

五、编译caffe




make pycaffe
make all
make test
make runtest


六、将PYTHONPATH写入.bashrc中




echo 'export PYTHONPATH=/home/<USER>/caffe/python' >> ~/.bashrc
source ~/.bashrc


七、在eclipse中使用


        将路径/home/<user>/caffe/python加入pydev插件中的python系统路径

附录:
http://caffe.berkeleyvision.org/ http://caffe.berkeleyvision.org/tutorial/ https://zhuanlan.zhihu.com/p/22129880 http://installing-caffe-the-right-way.wikidot.com/start https://github.com/BVLC/caffe/wiki/Ubuntu-16.04-or-15.10-Installation-Guide
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: