NVIDIA Jetson Xavier NX：安装Pytorch教程

1. 安装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版本。

2. 配置python环境

 sudo apt-get install python3-pip

 sudo pip3 install -U pip testresources setuptools

3. 安装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

（目前已经有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 ）

下载完成拷贝到NX后，进行安装，注意文件存放位置和执行命令的位置相一致，输入命令：

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

4. 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下载，然后编译安装。

直接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加清华源下载。

5. 安装成功。

请查看https://blog.csdn.net/qq_37301003/article/details/107139218实时更新