要进行深度学习的研究工作,就不得不提到Caffe,一个优秀的深度学习框架。
那我们应该如何安装Caffe呢?其实Caffe的官网上提供了非常好的安装教程,但所对应的Ubuntu版本为14.04,12.04,没有16.04版的安装教程(大哭),本篇博文就是教大家在Ubuntu16.04上安装Caffe的。
那我们要准备什么呢?所需软件:Oracle VM VirtualBox,Ubuntu16.04。
1.首先我们需要在VirtualBox中安装Ubuntu16.04,傻瓜式操作。
2.系统安装完成后,打开命令行控制台,输入以下命令,安装基本依赖(General dependencies)。
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
注意:由于网络情况的不同,一次安装可能不能全部安装成功,重复输入上述命令,再次安装(本人用了两次),这里建议大家一次安装一个软件包,便于弄清是那个未安装成功,以节省时间。
3.安装ATLAS,输入下述命令:
sudo apt-get install libatlas-base-dev
4.安装剩余依赖:
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
5.下载Caffe:
sudo apt-get install git
git clone https://github.com/BVLC/caffe.git
6.修改Makefile.config:
cp Makefile.config.example Makefile.config
gedit Makefile.config
找到#CPU_ONLY := 1,取消注释(我们设置为CPU模式)
找到
# Whatever else you find you need goes here.修改为
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib
# Whatever else you find you need goes here.INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serialLIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/i386-linux-gnu/hdf5/serial7.进行编译
make all
make test
make runtest
8.耐心等待编译完成
好了,到此为止,Caffe就安装完成了。尽情使用吧!