基于Jetson TX1的caffe环境搭建

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