Caffe在TX2下的安装方式与在一般的x86架构的电脑相同,唯一不同的是使用的文件全部都是arm64架构的。此外Caffe在当前使用JetPack3.1得到的是opencv2.4的版本,因此要注意依赖包的相关情况。
安装步骤如下:
安装一些必要的依赖项
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install --no-install-recommends libboost-all-dev
注意: 如果第一个apt-get产生了一些unable to install some broken file类似的话,记得将显示无法安装的包使用aptitude进行安装,血的教训,仅仅用aptitude安装那些无法安装的包来解析这个问题。
安装BLAS依赖项
sudo apt-get install libatlas-base-dev
安装Python及其依赖项
sudo apt-get install python
sudo apt-get install python-dev
sudo apt-get install python-numpy
sudo apt-get install ipython
sudo apt-get install ipython-notebook
sudo apt-get install python-sklearn
sudo apt-get install python-skimage
sudo apt-get install python-protobuf
谷歌glog和gflags和lmdb依赖项
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
安装git,并下载代码
sudo apt-get install git
git clone https://github.com/BVLC/caffe.git
进入源码目录进行caffe的编译准备
cd caffe
cp Makefile.config.example Makefile.config
根据需要修改Makefile.config
CUDA 计算能力警告
如果在编译时产生了CUDA计算能力的警告,解决的方式是将Makefile.config文件的CUDA_ARCH的前两行去掉。
./include/caffe/util/hdf5.hpp:6:18: fatal error: hdf5.h: No such file or directory
在Makefile.config文件中,添加 /usr/include/hdf5/serial/ 到 INCLUDE_DIRS,也就是把下面第一行代码改为第二行代码。
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial/
在Makefile文件中,把 hdf5_hl 和hdf5修改为hdf5_serial_hl 和 hdf5_serial,也就是把下面第一行代码改为第二行代码。
LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_hl hdf5
LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_serial_hl hdf5_serial
修改完成之后则前期的准备工作就大功告成了,最后的一步就是进行编译啦~
使用如下命令:
make all -j8
编译成功之后就可以愉快地玩耍啦~