环境

TX2环境：
jetpack4.5.1
cuda10.2
需要安装的版本：
torch1.8.0
torchvision0.9.0

安装

torch

https://forums.developer.nvidia.com/t/pytorch-for-jetson-version-1-10-now-available/72048
下载v1.8.0
sudo pip3 install torch-1.8.0-cp36-cp36m-linux_aarch64.whl +国内源

torchvision

https://github.com/pytorch/vision/tree/v0.9.0
copy to ubuntu
unzip
cd vision
sudo python3 setup.py install