在Jetson TX2上傻瓜式安装pytorch

0.前言

将Jetson TX2刷机完毕之后因为要用到pytorch，所以就在网上查找如何安装，但试了好多方法都失败了，然后找到了NVIDIA官网的安装教程，网址：nvidia官方。

1.步骤

（1）查看版本是否符合

上图是从所给网站上截下来的，可以看到支持安装JetPack4.2及以上版本的Nano,TX2,Xavier。而我的是安装有JetPack4.4 DP的TX2，是符合的。

（2）选择合适的Pytorch版本
从上面可以看到Pytorch v1.5.0和Pytorch v1.4.0都支持JetPack4.4 DP，因为之前安装了python3和pip3，所以我第一次选择了Pytorch v1.5.0，但后来出错了，如下：
显示Segmentation fault,这个错误的原因是torch的版本选择错误，所以我又把torch卸载，然后选择了Pytorch v1.4.0，之后就没有出现这种错误了。

（3）安装torch

按照上面的指示，在板子的终端依次输入Python3.6版本下的命令来安装torch，但这里很可能会出问题，wget https://nvidia.box.com/shared/static/ncgzus5o23uck9i5oth2n8n06k340l6k.whl -O torch-1.4.0-cp36-cp36m-linux_aarch64.whl这个命令可能执行不了，因为国内访问不了这个地址，这时候就要翻墙了，如果电脑上有翻墙软件直接翻墙即可访问下载。如果电脑上没有的话也可以用手机翻墙访问第（2）步中要下载的选项先下载下来然后放到板子的home里面，然后执行

sudo apt-get install python3-pip libopenblas-base libopenmpi-dev 
pip3 install Cython
pip3 install numpy torch-1.4.0-cp36-cp36m-linux_aarch64.whl

即可。
如果顺利的话就可以安装好torch了。

（4）安装torchvision

安装torchvision的第二步要git一个版本的torchvision，而这个版本要参考上图，对应Pytorch v1.4选择了trochvision v0.5.0，命令如下：

sudo apt-get install libjpeg-dev zlib1g-dev
git clone --branch v0.5.0 https://github.com/pytorch/vision torchvision   # see below for version of torchvision to download
cd torchvision
sudo python setup.py install
cd ../  # attempting to load torchvision from build dir will result in import error

顺利的话就成功了。

（4）测试

以上命令能正常输出就通过测试了。

TX2的刷机方法