TX2上安装pytorch

由于TX2是arm架构，无法直接通过官方的方法——pip或conda等安装，需要自行编译github上的软件包才能使用。

注：本文略过TX2上的CUDA，Cudnn以及opencv的安装，这些部分的安装大家可以自行百度。

现在正式开始编译pytorch吧：

1.打开终端，输入以下命令下载git包

git clone https://github.com/pytorch/pytorch.git

下载完后，大家会看到在home下有了pytorch文件夹。

2.切分版本

cd到home下的pytorch文件夹，进行版本切分，我要使用pytorch0.3.1版本，所以依次输入以下命令进行切分：

cd '/home/nvidia/pytorch'

git checkout -b v0.3.1 v0.3.1

3.更新submodule

此步不进行，在install的时候会报Could not find /home/nvidia/pytorch/torch/lib/gloo/CMakeLists.txt错误

在pytorch文件夹路径下，在终端输入以下命令进行更新：

git submodule update --init

4.正式编译安装

还是在当前pytorch文件夹路径下，输入以下命令进行编译安装：

sudo python setup.py install

切记一定要带上sudo，不然编译很久很久要完成后，你会发现报Permission denied的错误。

（编译时间很长，大家请耐心等待，睡一觉就好了）

5.测试

在python下输入import torch，没有显示错误则说明编译安装成功，如下图：