for Jetson SDK 4.4 we’ll install Pytorch v 1.8.0

for checking Jetson SDK version use command:

git clone https://github.com/jetsonhacks/jetsonUtilities.git cd jetsonUtilities python jetsonInfo.py

Below are commands for installing these PyTorch wheels on Jetson.

> Python 3.6

wget <https://nvidia.box.com/shared/static/p57jwntv436lfrd78inwl7iml6p13fzh.whl> -O torch-1.8.0-cp36-cp36m-linux_aarch64.whl
sudo apt-get install python3-pip libopenblas-base libopenmpi-dev libomp-dev
pip3 install Cython<3
pip3 install numpy torch-1.8.0-cp36-cp36m-linux_aarch64.whl

For installing torchvision.

> torchvision

sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libopenblas-dev libavcodec-dev libavformat-dev libswscale-dev
git clone --branch v0.9.0 <https://github.com/pytorch/vision> torchvision# see below for version of torchvision to download 
cd torchvision
export BUILD_VERSION=0.9.0 # where 0.x.0 is the torchvision version
python3 setup.py install --user
cd ../# attempting to load torchvision from build dir will result in import error
pip install 'pillow<7'# always needed for Python 2.7, not needed torchvision v0.5.0+ with Python 3.6

for PyTorch v1.8 we will install torchvision v0.9.0