Jetson TX2 安装Caffe

首先说下我的TX2的系统是Ubuntu 18.04。系统里安装的由Python3.6和Python 2.7.

在我安装Caffe官方给我的步骤http://caffe.berkeleyvision.org/install_apt.html，因为TX2的系统是aarch64的，所以无法按照官方说的直接sudo apt install caffe-cuda，所以需要下载源码编译，编译以后发现Python3无法import caffe，网上说的方法很多了，但是在我的TX2上面都没有用，下面我来说说我的解决办法。

首先我想在Python3上面直接import，所以需要把Makefile.config里面的68和69行

PYTHON_INCLUDE := /usr/include/python2.7 \
        /usr/lib/python2.7/dist-packages/numpy/core/include

注释掉，改为

#PYTHON_INCLUDE := /usr/include/python2.7 \
#       /usr/lib/python2.7/dist-packages/numpy/core/include

然后把78,79,80行

# PYTHON_LIBRARIES := boost_python3 python3.5m
#PYTHON_INCLUDE := /usr/include/python3.5m \
#                 /usr/lib/python3.5/dist-packages/numpy/core/include

由于我的Python3版本为3.6,所以改为

 PYTHON_LIBRARIES := boost_python3 python3.6m
 PYTHON_INCLUDE := /usr/include/python3.6m \
                 /usr/lib/python3.6/dist-packages/numpy/core/include

第二个就是需要把CMakeLists.txt的第35行

set(python_version "2" CACHE STRING "Specify which Python version to use")

改为

set(python_version "3" CACHE STRING "Specify which Python version to use")

接下来就是和官方介绍的一样了，进入caffe的目录，

mkdir build
cd build
cmake ..
make -j8 all
make install
sudo gedit ~/.bashrc

然后将 export PYTHONPATH=~/caffe/python:$PYTHONPATH

上面的路径需要根据自己caffe的所在目录修改，可进入caffe/python目录然后敲pwd命令查看路径

加到做后一行，保存退出。

source ~/.bashrc

可以在终端里面试一下

python3

import caffe

如果没有问题就是成功了