本文介绍 Caffe 在 Jetson TX1 開發平台上安裝與使用深層神經網路函式庫 (cuDNN). 
使用套件如下:

JetPack For L4T 2.1 (L4T 23.2, Ubuntu 14.04 LTS)
CUDA Toolkit 7.0
cuDNN V4 (ARMv7)
Caffe

你可能需要登入 NVIDIA 開發者網站 (https://developer.nvidia.com) 才能下載 CUDA 與 cuDNN 套件. JetPack For L4T 2.1 安裝不在此說明.

Jetson TX1 Ubuntu 14.04 環境設定

$ sudo add-apt-repository universe
$ sudo apt-get update
$ 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 libgflags-dev libgoogle-glog-dev liblmdb-dev libblas-dev liblapack-dev libatlas-base-dev


CUDA Toolkit 安裝

$ wget http://developer.download.nvidia.com/embedded/L4T/r23_Release_v2.0/CUDA/cuda-repo-l4t-r23.1-7-0-local_7.0-73_armhf.deb
$ sudo dpkg -i cuda-repo-l4t-r23.1-7-0-local_7.0-73_armhf.deb 
$ sudo apt-get update
$ sudo apt-get install cuda-toolkit-7-0

cuDNN 安裝

從 https://developer.nvidia.com/rdp/cudnn-download 下載 cuDNN. 本文使用 cuDNN v4 Library for L4T (ARMv7).

$ tar -zxvf cudnn-7.0-linux-ARMv7-v4.0-prod.tgz

$ sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
$ sudo cp cuda/lib/* /usr/local/cuda/lib/


Caffe 下載

Caffe 原始碼可由 GitHub 專案取得.

$ git clone https://github.com/BVLC/caffe.git

Caffe 配置與编譯

$ cd caffe
$ cp Makefile.config.example Makefile.config
$ gedit Makefile.config
使用 cuDNN
# cuDNN acceleration switch (uncomment to build with cuDNN).
USE_CUDNN := 1

$ export LD_LIBRARY_PATH=/usr/local/cuda/lib/
$ make -j4 all

測試

$ make -j4 test
$ make runtest

(完)

文章標籤

創作者介紹
創作者 版大 的頭像
版大

嵌入式系統

版大 發表在 痞客邦 留言(0) 人氣()