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. 安裝成功。