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的環境搭建了