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刷机过程;
完
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:【软件安装与环境配置】TX2安装配置caffe过程 - Python技术站