Jetson TX2-JetPack 4.3 安装 pytorch和torchvision

声明：

TX2系统： Ubuntu18.04
python版本：3.6.9

刷入JetPack 4.3，apt更换为国内源。选择这个版本的原因是对应cuda的版本为10.0。（JetPack 4.4对应cuda版本为10.2，如果需要cuda 10.2版本，请刷入4.4。）

---

本文基于JetPack4.3，安装pytorch版本为1.2，torchvision版本为0.4.0

1、安装pytorch 1.2

pytorch1.2.0 CSDN下载链接，下载完之后运行：

pip3 install torch-1.3.0-cp36-cp36m-linux_aarch64.whl

2、安装torchvision 0.4.0
首先安装libjpeg-dev zlib1g-dev

sudo apt install libjpeg-dev zlib1g-dev

然后安装pillow，注意顺序

pip3 install 'pillow<7'

最后安装torchvision，版本一定要是0.4.0

git clone --branch v0.4.0 https://github.com/pytorch/vision torchvision 
cd torchvision
export BUILD_VERSION=0.4.0 
sudo python setup.py install 

完