Tx2刷机

 

sudo sh ./caffe_dependency.sh

 注意python的版本问题。

问题

LD -o .build_release/lib/libcaffe.so.1.0.0
/usr/bin/ld: cannot find -lhdf5_hl
/usr/bin/ld: cannot find -lhdf5
collect2: error: ld returned 1 exit status
Makefile:584: recipe for target '.build_release/lib/libcaffe.so.1.0.0' failed
make: *** [.build_release/lib/libcaffe.so.1.0.0] Error 1

参考here

# This code is taken from https://github.com/sh1r0/caffe-android-lib
ifeq ($(USE_HDF5), 1)
#LIBRARIES += hdf5_hl hdf5
LIBRARIES += hdf5_serial_hl hdf5_serial
endif

最后可以编译成功。

运行mnist的example时出现错误

dia@tegra-ubuntu:~/caffe$ ./examples/mnist/train_lenet.sh
I0321 13:57:34.925858 26097 caffe.cpp:204] Using GPUs 0
I0321 13:57:34.935622 26097 caffe.cpp:209] GPU 0: NVIDIA Tegra X2
F0321 13:57:35.523316 26097 benchmark.cpp:112] Check failed: error == cudaSuccess (30 vs. 0) unknown error
*** Check failure stack trace: ***

怀疑是GPU驱动的问题,与别人交流,说是TX2刷机的时候已经内置的有GPU,验证CUDA实例也证明CUDA正确安装,显示驱动是NVIDIA Tegra2。不过,不能用于caffe的训练等,不知道是不是这样,也没有查到相关信息。

准备试一下CPU版本的caffe,重新编译之后,已经开始训练mnist啦。。

 

 

参考

1. Jetson TX2学习笔记(二):caffe安装配置;

2. TX2刷机过程;