最近一直在业余时间维护服务器,给多台服务器配上了caffe供训练用。
本文记录了12月1号在Ubuntu server 14.04上安装当时最新版caffe的全过程。
另外注意BLAS使用的是ATLAS,刚开始试了openblas装上引用不了,试了些文章里的方法也不行,就没详细研究。
想装openblas的这一部分可以参考别的文章,官网说openblas会比ATLAS快一些。
先给出caffe官网以及caffe的github:
1.caffe官网:http://caffe.berkeleyvision.org/
2.caffe官网的ubuntu安装指导页面:http://caffe.berkeleyvision.org/install_apt.html
3.caffe的github:https://github.com/BVLC/caffe
一、前提
机器上已经配好cuda8.0,NVIDIA显卡驱动,opencv2.4.8以及python2.7。
若你的机器上没有请先参考别的文章把这几样装好。
二、一些准备工作
1.由于ubuntu新机器的源往往是从美国服务器下载,使用apt-get下载时,速度不能再慢,建议先改为国内的源,如163,阿里,搜狐,上海交大等,以免后面安装太慢影响进度和心情。这点网上有许多说明,在此不再赘述,给出一个参考链接:
http://blog.csdn.net/xukai871105/article/details/24887245
2.
三、安装步骤
0.安装所有依赖
参考caffe官网的ubuntu安装指导页面。
CUDA、NVIDIA驱动和opencv假设已经装好。
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
BLAS
sudo apt-get install libatlas-base-dev
Python
sudo apt-get install python-dev
Remaining dependencies, 14.04
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
1. 下载caffe
在ubuntu终端使用git clone命令从caffe官网下载caffe:
git clone https://github.com/BVLC/caffe.git
下载到任何地方都可以,不太影响后面的操作。
2.编译caffe
用linux的cd命令打开到caffe所在目录(没改过的话就是caffe-master)下面,生成Makefile.config
cp Makefile.config.example Makefile.config
Makefile.config不用改(当然你流程不一样的话可以自己改一下),
输入make命令编译caffe:
make all -j16
make test -j16
make runtest -j16
其中-j16是指16个线程,也可以根据自己电脑的核数自行安排。
如果中间出错了,需要先清除编译再重新执行:
make clean
3.配置python依赖并编译、配置环境变量
这一步可以参考如下链接,比较全面:
http://blog.csdn.net/tsinghuahui/article/details/46790705
把其中的命令复制过来如下:
sudo apt-get install python-numpy python-scipy python-matplotlibpython-sklearn python-skimage python-h5py python-protobuf
sudo apt-get update
make pycaffe
注意其中的python-h5py命令是适用于hdf5文件的。如果你要用lmdb或者leveldb来使用caffe文件,则需参考别的文章安装相应python依赖。
先用env命令查看原来是否有PYTHONPATH,一般是没有的
env如果有,在用户的.bashrc文件最后一行添加:
export PYTHOPATH = $PYTHONPATH:/caffe所在目录/python
如果没有,在用户的.bashrc文件最后一行添加:
export PYTHOPATH = /caffe所在目录/python
.bashrc文件一般在用户目录下,如root就在root目录下,其他就在/home/你的用户名/ 下面,注意开启隐藏文件
4.查看hdf5文件
装一个Linux下的扩展包hdfview用于查看hdf5文件:
apt-get install hdfview
好的!恭喜!
到此为止,就可以使用caffe了!
有什么问题请在下面留言!