最近博主要准备着手用caffe进行人工智能方向的学习,殊不知败在了caffe环境搭建上,历经千辛万苦不断的Google终于把caffe搭建好了,下面就给出博主搭建caffe方法。
如果你是刚刚装好虚拟机,为了防止下载速度过慢,请用以下方法将Linux的下载源改为国内源。
1.登录系统
2.修改软件源:APT-sources.list
打开终端输入:
sudo gedit /etc/apt/sources.list
3.选择适合自己较快的源:
源链接:http://wiki.ubuntu.org.cn/index.php?title=%E6%BA%90%E5%88%97%E8%A1%A8&variant=zh-cn
4.对软件进行一次整体更新:
sudo apt-get update
sudo apt-get upgrade
安装依赖包
1.安装protobuf,leveldb,snappy,opencv,hdf5, protobuf compiler andboost:
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
2.安装gflags,glogs ,lmdb andatlas.
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
sudo apt-get install libatlas-base-dev
下载Caffe
使用git直接下载Caffe非常简单,或者去https://github.com/BVLC/caffe下载
sudo apt-get install git
git clone git://github.com/BVLC/caffe.git
编译Caffe
1.切换到Caffe所在目录
cp Makefile.config.example Makefile.config
配置Makefile.config
1)CPU_ONLY := 1
2)配置一些引用文件(增加部分主要是解决新版本下,HDF5的路径问题)
-两个路径要改成这样:(添加后面的两个hdf5的路径, 否则编译时报hdf5错误)
/# 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/hdf5/serial
3)设置Python Caffe(将上面的路径改为下面的路径)
PYTHON_INCLUDE := /usr/include/python2.7 \
/usr/lib/python2.7/dist-packages/numpy/core/include**
PYTHON_INCLUDE := /usr/include/python2.7 \
/usr/local/lib/python2.7/dist-packages/numpy/core/include
编译python接口
1.Caffe拥有python\C++\shell接口,在Caffe使用python特别方便,在实例中都有接口的说明。
1)确保pip已经安装
sudo apt-get install python-pip
2)切换到Caffe/python所在目录,新建shell文件并执行安装依赖
for req in $(cat requirements.txt); do pip install $req; done
Make Caffe
sudo make pycaffe -j8
sudo make all -j8
sudo make test -j8
sudo make runtest -j8
博主在Make Caffe这一步遇到了一万个错误,一般直接把错误信息复制到StackOverflow上都可以得到解答,一定要有耐心,相信自己一定可以编译通过的!
另外, 这个make默认是用CPU单核运算,如果想要快一点, 比如我想使用八核, 在make后面加上-j8标签.如果上面4行某一行报错之后想要重试,一定要先sudo make clean再重新开始
这时候应该可以了,试验一下:
切换到Caffe/python所在目录
$ python
Python 2.7.12 (default, Jul 1 2016, 15:12:24)
[GCC 5.4.0 20160609] on linux2
Type "help", "copyright", "credits" or "license" for more information.
>>> import caffe
>>>
说明安装全部完成!
参考:
[1]http://blog.csdn.net/u010402483/article/details/51506616 Caffe学习(1):Ubuntu16.04上Caffe配置安装(Only CPU)
[2]http://www.linuxidc.com/Linux/2016-09/135034.htm Ubuntu 16.04上安装Caffe(CPU only)
[3]http://blog.csdn.net/weixin_38034182/article/details/76672906 Linux系统软件安装更新下载太慢解决方法(更换国内源)