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

我需要在caffe上运行深度学习程序,因此在这里把caffe GPU在jetson tx2(下称TX2主机)安装的过程记录下来。

由于JetPack中自带CUDA、OpenCV和CuDNN,在完成TX2主机JetPack安装后(教程可参见我的前一篇博文https://blog.csdn.net/u012614287/article/details/81509677),安装caffe for python(GPU)的过程比在一般PC安装的过程简单一些。主要可概括为以下几步:1.安装依赖项。2.从git上下载caffe源码。3.根据需要修改配置文件。4.编译源码。5.针对python安装部分caffe的python接口。详述见下文。

一、安装依赖项。
caffe有许多依赖项,这里我不加解释的将其列出,这步宁多勿缺,未曾有问题出现。在TX2主机的ubuntu16.04环境中可以将这些代码写在脚本文件中一次运行,以下是我安装依赖脚本文件(dependency.sh)的内容。

# general dependency
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

# BLAS dependency
sudo apt-get install libatlas-base-dev

# python dependency
sudo apt-get install python
sudo apt-get install python-dev

# python dependency etc.
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

# google dependency
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

接下来输入命令./dependency.sh执行该脚本,等待依赖安装完成。至此依赖安装结束

二、从git上下载caffe源码。
如果TX2主机上没有事先安装git,可以通过以下命令安装。

sudo apt-get install git

接下来可以cd至希望caffe源码保存的目录,使用以下命令将caffe源码down下来。

git clone https://github.com/BVLC/caffe.git

源码安装完成

三、根据配置需要修改配置文件。
在开始编译之前,需要进入caffe目录修改’Makefile.config’和’Makefile’两个配置文件。

Makefile.config文件通过复制Makefile.config.example文件得到,根据我的需求,对该文件做了以下改动
1.使用CuDNN(取消USE_CUDNN := 1前的‘#’)
2.使用OpenCV v3(取消OPENCV_VERSION := 3前的‘#’)
3.由于使用的是CUDA9版本,按要求修改了代码(删除了CUDA_ARCH变量的前两行,具体可看配置文件里该变量的前几行)。
4.使用python接口(取消WITH_PYTHON_LAYER := 1变量前的’#’)
5.修改python路径如下:

INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu /usr/lib/x86_64-linux-gnu/hdf5/serial 

Makefile文件作出的改动如下:

将:
NVCCFLAGS +=-ccbin=$(CXX) -Xcompiler-fPIC $(COMMON_FLAGS)
替换为:
NVCCFLAGS += -D_FORCE_INLINES -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS)
将:
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

配置文件修改完成

四、编译源码
caffe源码目录下执行编译命令

make all -j8

运行测试

sudo make test -j8
sudo make runtest -j8

如果没有异常,则caffe安装完成,但由于没安装pycaffe接口,依然无法在python中导入caffe。
此时需要在~/.bashrc文件中写入caffe的路径,在文件下方写入

export PYTHONPATH=~/caffe/python:$PYTHONPATH

然后使用source ~/.bashrc使环境变量生效即可。至此caffe for python在TX2主机安装完成!

这里写图片描述

猜你喜欢

转载自blog.csdn.net/u012614287/article/details/81537743