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















發佈了31 篇原創文章 · 獲贊 20 · 訪問量 9萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章