刷机教程参照:http://blog.csdn.net/Jalong_Ma/article/details/52743923里面的介绍即可。
caffe配置:
caffe的配置可能遇到很多问题,我推荐的步骤是:
参照http://blog.csdn.net/Jalong_Ma/article/details/52743923安装依赖库,然后:
安装NCCL:
$ git clone https://github.com/NVIDIA/nccl.git
$ cd nccl
$ sudo make install -j4
NCCL 库和文件头将安装在 /usr/local/lib 和 /usr/local/include 中。
git clone https://github.com/BVLC/caffe.git
cd caffe
cp Makefile.config.example Makefile.config
修改
Makefile.config
:
# cuDNN acceleration switch (uncomment to build with cuDNN).
USE_CUDNN := 1
# Uncomment if you're using OpenCV 3
OPENCV_VERSION := 3
# Uncomment to support layers written in Python (will link against Python libs)
WITH_PYTHON_LAYER := 1
此外还需要:
把#USE_NCCL := 1改为: USE_NCCL := 1
TX1的GPU计算能力是53,所以更改Makefile.config文件中更改:CUDA_ARCH 下加入一行:-gencode arch=compute_53,code=sm53 \
由于hdf5库目录更改,所以需要单独添加:
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/
(单独添加的意思就是把上面这两行复制到MaKefile.config文件里即可)。
之后便可以编译:
make -j4 all
make -j4 runtest
.......按照上面的参考博客做就行。