Jetson TX1中Caffe安装

Jetson TX1是arm架构的与在电脑上的编译有所不同。建议使用32位的系统，可以避免不少坑。TX1目前出厂默认是64位系统，建议刷到32位系统，具体可参考我上一篇博客。

1. 配置安装依赖

$ sudo add-apt-repository universe
$ sudo apt-get update
$ sudo apt-get install cmake git aptitude screen g++ libboost-all-dev \
libgflags-dev libgoogle-glog-dev protobuf-compiler libprotobuf-dev \
bc libblas-dev libatlas-dev libhdf5-dev libleveldb-dev liblmdb-dev \
libsnappy-dev libatlas-base-dev python-numpy libgflags-dev \
libgoogle-glog-dev python-skimage python-protobuf python-pandas \
libopencv-dev

2. 下载caffe

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

3. 编译caffe

在编译前需要对caffe的配置文件进行修改

$ cd caffe  
$ mv Makefile.config.example Makefile.config  
$ gedit Makefile.config  

• 第5行：去掉#号，即 “USE_CUDNN := 1”
• 修改CUDA_ARCH
  CUDA_ARCH := -gencode arch=compute_53,code=compute_53
  其他用＃注释掉。
  这里53是TX1的Compute Capability，具体可以在NVidia网站查询，TX1是5.3，计算性能是非常高的。这里一定要准确设置，避免后面很多错误和坑。
• 保存退出

编译caffe

$ make all -j4  

4. 用caffe自带的mnist例子测试

注意新版caffe例子的运行需要在caffe目录下，不要进入到example目录去执行脚本，会出现找不到文件的错误。

./data/mnist/get_mnist.sh
./examples/mnist/create_mnist.sh
./examples/mnist/train_lenet.sh 

如果上述指令运行没有错误，则安装成功。