ubuntu安装cpu版caffe

  • Post author:
  • Post category:其他


低配电脑搞神经网络,用虚拟机实验一波!官方文档十分详细:

点击打开链接

以下是安装过程:

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

由于ubuntu的库有各种依赖关系,apt-get可能无法解决,建议使用aptitude,会给出多个解决方案,实测可行!sudo aptitude install …

2、因为是虚拟机,无法使用gpu,可以跳过安装cuda!(而且好像16.04已经带有cuda8)

3、安装ATLAS

sudo apt-get install libatlas-base-dev

4、下载caffe

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

5、修改Makefile.config

cd caffe
cp Makefile.config.example Makefile.config 
gedit Makefile.config

将# cpu_only := 1的注释去掉,找到并修改为:

INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/i386-linux-gnu/hdf5/serial

6、编译安装

make all
make test
make runtest

到此caffe安装已经完成!

若有需要用到python或matlab接口的,先设置好Makefile.config中的路径,再另外编译:

make pycaffe
make matcaffe

添加python环境变量,方便以后imoprt caffe,打开/etc/profile末尾添加:

PYTHONPATH=/xxx/xxx/caffe/python:$PYTHONPATH

另外pycaffe的接口暴露在caff目录下的python文件夹,只需要import caffe就可以直接调用。

matcaffe接口官网有介绍。

感谢

http://blog.csdn.net/oyjxer/article/details/51824254

提供的帮助!



版权声明:本文为Brunoon5原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。