Caffe的安装及配置

来源:互联网 发布:迈尔斯特纳数据 编辑:程序博客网 时间:2024/06/04 18:41

Ubuntu上Caffe的安装及配置

系统:Ubuntu 17.04

1.安装依赖项

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

2.安装必要的库

sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev  sudo apt-get install libatlas-base-dev  

3.下载caffe

cd ~   git clone https://github.com/BVLC/caffe.git #克隆caffe到本地,并命名为caffe

4.配置caffe

4.1生成Makefile.config文件

cd caffe/cp Makefile.config.example Makefile.config

4.2修改Makefile.config文件中的配置

sudo gedit Makefile.config#第一步#去掉CPU_ONLY:=1的注释#第二步#将下面第一行代码改为第二行代码#INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include#INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial/

5.编译caffe

make allmake pycaffemake distributemake test

若出现错误“fatal error: hdf5.h: 没有那个文件或目录”,4.1.2中第二步为解决方法。

6.配置caffe的pyhton接口pycaffe

6.1安装接口依赖库

sudo apt-get install python-pipsudo atp-get install python-dev python-numpysudo apt-get install gfortransudo pip install –r python/requirements.txtsudo pip install pydotnumpy scipy matplotlib sklearn skimage h5py protobuf leveldb networkx nose pandas gflags cython ipython gfortran

验证

安装结束后,可以执行如下语句验证:

sudo pip install -r requirements.txt

如果显示Requirement already satisfied,则安装成功,否则会继续进行安装。

6.2将caffe根目录下的python文件夹加入到环境变量中

vi ~/.bashrcexport PYTHONPATH=/usr/caffe/python:$PYTHONPATH#在打开的文档中加入以下代码export PYTHONPATH=/home/lynn/caffe/python:$PYTHONPATH#执行更新配置sudo ldconfig

6.3编译Python接口

 cd ~/caffe/ sudo make pycaffe

如果出现找不到numpy/arrayobject.h这种问题,则检查Makefile.config文件中的PYTHON路径(Python.h、
numpy/arrayobject.h的路径):
PYTHON_INCLUDE := /usr/include/python2.7 /usr/lib/python2.7/dist-packages/numpy/core/include

6.4验证python接口

进行python环境,引入caffe包,如果没有报错,则安装成功!如下图:
安装成功

原创粉丝点击