NVIDIA Jetson TX1刷机教程与caffe配置(有点潦草,个人觉得有帮助)

刷机教程参照: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 中。


编译caffe:
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
.......按照上面的参考博客做就行。




评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值