NVIDIA Jetson NX刷机安装pytorch（看这一篇就够了！！！）

1. 下载系统
登陆https://developer.nvidia.com/embedded/downloads
本人下载的版本为 JP 4.4 DP，2020/05/14发布，大小6GB左右。
连接：https://share.weiyun.com/0If2NL1x 密码：drbgim
2. 下载烧录工具
连接：https://pan.baidu.com/s/1Gry-YhI_tHV-N_SmFoql1Q
提取码：08du
插入读卡器，选择刚刚下载的系统，第二步软件会自动识别，而后点击flash，等待完成。
3.配置系统
插入到NX上，插上硬件，开机，进入到系统配置的界面，须要注意的地方是配置最好选择英文，地点选择中国上海，用户名密码全填nvidia。而后默认选项一路到最后重启。
4. 安装python环境以及依赖包

sudo apt-get update
sudo apt-get upgrade
sudo apt-get install git cmake python3-dev
sudo apt-get install libhdf5-serial-dev hdf5-tools libhdf5-dev zlib1g-dev zip libjpeg8-dev

备注：本人默认下载源，换源可否成功等待之后测试，该步时间须要大概三四个小时。JP4.4版本的cuda版本为10.2，因此必须pytorch1.5版本才能够，系统自带cudnn，这些无需配置。python版本为2.7版本和3.6版本，本文没有更改python的软连接，默认python为2.7，用命令python3和pip3来调用python3.6版本。
5. 配置python环境

sudo apt-get install python3-pip
 sudo pip3 install -U pip testresources setuptools

6. 安装pytorch
先安装依赖：

sudo apt-get install libopenblas-base libopenmpi-dev

sudo pip3 install mpi4py
sudo pip3 install Cython

下载pytorch编译好的.whl文件
连接：https://pan.baidu.com/s/1QE7f9TiE-dXaodh895wnSw
提取码：kr2u
下载完成拷贝到NX后，进行安装，注意文件存放位置和执行命令的位置相一致，输入命令：

sudo pip3 install torch-1.5.0-cp36-cp36m-linux_aarch64.whl

7. torchvision安装

sudo apt-get install libjpeg-dev zlib1g-dev
 git clone --branch  v0.6.0 https://github.com/pytorch/vision torchvision

直接pip3下载是没有torchvision0.6版本，须要git下载，而后编译安装。

cd torchvision
 sudo python3 setup.py install

安装到最后须要pillow包，最后可能会下载失败。可是torchvision已经安装成功了，输入命令pip3 list能够看到。
而后pip3 install pillow安装pillow包便可，若是下载速度慢，能够加入-i加清华源下载。
8. 安装成功。