工具准备
sudo apt-get update
sudo apt-get upgrade
sudo apt-get install -y build-essential
sudo apt-get install -y cmake
sudo apt-get install -y git
sudo apt-get install -y pkg-config
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
以上安装过程如果出现中断或者失败,请重试几次即可。
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
sudo apt-get install libatlas-base-dev
sudo apt-get install python-dev
下载caffe
git clone https://github.com/BVLC/caffe.git
如果pip尚未安装,请安装pip以后继续。
sudo apt-get install python-pip
安装python必要库
cd caffe
cat python/requirements.txt | xargs -L 1 sudo pip install
网络一般比较慢,请耐心完成安装,如果一次不成功,请多试几次。
添加软链
sudo ln -s /usr/include/python2.7/ /usr/local/include/python2.7
sudo ln -s /usr/local/lib/python2.7/dist-packages/numpy/core/include/numpy/ /usr/local/include/python2.7/numpy
配置caffe
进入上面刚下载的caffe目录,今后的操作都在这个目录下面进行。
复制
Makefile.config.example
作为Makefile.config
,如下图所示。
然后编辑该配置文件,如下图所示。
修改include的路径为本地
/usr/local/lib/python2.7/dist-packages/numpy/core/include
,另外修改hdf5的路径。
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib
改为
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 /usr/lib/x86_64-linux-gnu/hdf5/serial
如下图所示。
编译caffe
make pycaffe
make all
make test
make testrun
出现如下图所示,表示编译完成。
测试caffe
- 直接在python中import caffe看是否报错是最简单的验证方式
然而,让人不爽的是居然报错了。
解决办法很简单,只需要在环境变量中加入caffe所在python目录即可。
- 下载运行MNIST栗子这里给个简单的提示,不再给出最后结果
cd ~/caffe #将终端定位到Caffe根目录
./data/mnist/get_mnist.sh #下载MNIST数据库并解压缩
./examples/mnist/create_mnist.sh #将其转换成Lmdb数据库格式
编辑lenet_solver的solver_mode模式从GPU改为cpu:
vim ./examples/mnist/lenet_solver.prototxt
训练模型
cd $CAFFE_ROOT
./examples/mnist/train_lenet.sh
当然也可以下载imagenet的caffe模型和label
./scripts/download_model_binary.py models/bvlc_reference_caffenet
./data/ilsvrc12/get_ilsvrc_aux.sh
具体如何使用,这里不再赘述。
参看文档:
虚拟机VB.Ubuntu16.04.1.无GPU caffe环境配置
虚拟机下Ubuntu安装CPU版本的caffe