Jetson tx2(JetPack 4.4)配置pytorch环境

下载pytorch

1. 下载pytorch1.7：我的系统是JetPack4.4，要求pytorch>=1.7.
   

安装pytorch

sudo apt-get install libopenblas-base libopenmpi-dev
pip3 install torch-1.7.0-cp36-cp36m-linux_aarch64.whl

安装torchvision

1. 这里选择torchvision 0.8.1（可根据自己的pytorch选择torchvision版本）

sudo apt-get install libjpeg-dev zlib1g-dev
git clone --branch v0.8.1 https://github.com/pytorch/vision torchvision   # see below for version of torchvision to download
cd torchvision
sudo python setup.py install