Data: 2017.03.13
Author: cjh
Theme: Set up the Caffe environment
步骤1:安装caffe所需的安装包
$ sudo add-apt-repository universe
$ sudo add-apt-repository multiverse
$ sudo apt-get update
$ sudo apt-get install libboost-all-dev libprotobuf-dev libleveldb-dev libsnappy-dev
$ sudo apt-get install libhdf5-serial-dev protobuf-compiler libgflags-dev libgoogle-glog-dev
$ sudo apt-get install liblmdb-dev libblas-dev libatlas-base-dev
步骤2:下载caffe
1. git clone git://github.com/BVLC/caffe.git
2.若网速不够可在windows上下载
https://github.com/BVLC/caffe
然后复制到你的开发板上如下命令
$ mkdir -pv $HOME/Work/caffe
$ cp caffe-master.zip $HOME/Work/caffe/
$ cd $HOME/Work/caffe/ && unzip caffe-master.zip
步骤3:编译caffe
$ cd $HOME/Work/caffe/caffe-master
$ cp Makefile.config.example Makefile.config
$ vim Makefile.config
修改Makefile.config文件
USE_CUDNN := 1
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/aarch64-linux-gnu/hdf5/serial
有必要课修改cuda的路径,我的开发板用的是cuda-8.0
所以我cuda的路径修改为了/usr/local/cuda-8.0
修改完成后保存并退出
$ make -j4
经过以上步骤应该就能完成caffe的环境搭建了