Dji Manifold2-G、Jetson TX2源码编译安装pytorch

Dji Manifold2-G、Jetson TX2源码编译安装pytorch

最近由于所承担项目的实际需求，需要在Dji Manifold2-G上面配置深度学习的网络框架pytorch。Dji Manifold2-G实际上是集成了一块NVIDIA Jetson TX2的GPU模块，并提供多种接口用来连接不同的外部设备，体积小，操作便捷，非常适合于作为机载的一种处理器。自己在实际的编译过程中遇到各种Bug，各种网上寻找解决方案，现在将自己的源码编译安装pytorch过程整理处理，方便以后需要的开发者们节省更多时间。

一、安装cudnn

源码编译安装pytorch的时候会使用cuda和cudnn，Dji Manifold2-G出厂自带的有cuda9.0，因此不需要安装cuda，只需要安装对应cuda9.0版本的cudnn，Dji Manifold2-G的CPU是基于ARM架构的，即对应的是aarch64，因此去官网下载的时候应该下载对应版本的cudnn，下面给出我自己使用版本的一个链接。

http://developer.download.nvidia.com/compute/redist/cudnn/v7.1.1/cudnn-9.0-linux-x64-v7.1.tgz

安装cudnn

~$ cd data/
~$ sudo tar xvf cudnn-9.0-linux-x64-v7.1.tgz
~$ mkdir cudnn7
~$ sudo cp -r cuda cudnn7
~$ cd cudnn7
~$ sudo cp include/cudnn.h /usr/local/cuda/include/
~$ sudo cp lib64/* /usr/local/cuda/lib64/
~$ cd /usr/local/cuda/lib64/
/usr/local/cuda