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版本。（目前已经有pytorch1.6版本，下载地址：https://forums.developer.nvidia.com/t/ pytorch  -for-jetson-version-1-6-0-now-available/72048或者百度网盘链接：https://pan.baidu.com/s/1zGbxO6xaDyQO3GpoUbjgqA

   提取码：f72u ）
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文件（1.5版本，1.6的请看第4步）

   链接：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以及0.7版本，需要git下载，然后编译安装。

   如果git下载失败，可以使用码云将https://github.com/pytorch/ vision  导入，然后如图选择，选择完成后进行下载。（建议使用该步骤，亲测可行！！！吐血推荐！！！）

cd进入到目录中，git命令下载时修改了文件名称，为torchvision（码云默认下载为vision，无修改）cd路径读者自行判断。

 cd torchvision
 sudo python3 setup.py install

安装到最后需要pillow包，最后可能会下载失败。但是torchvision已经安装成功了，输入命令pip3 list可以看到。

   然后pip3 install pillow安装pillow包即可，如果下载速度慢，可以加入-i加清华源下载。
8. 安装成功。

免责声明：本文系网络转载或改编，未找到原创作者，版权归原作者所有。如涉及版权，请联系删