How to install PyTorch and Torchvision on your Jetson device

From 1 and 2 install the required version of PyTorch (check example.sh)
To build Torchvision from source:

1. git clone --recursive --branch <branch_name i.e. version> https://github.com/pytorch/vision
   Note: Check the compatible Torchvision version for your PyTorch version from 2
2. cd vision
3. export FORCE_CUDA=1 # Only for devices with CUDA
4. MAX_JOBS=1 python setup.py install # MAX_JOBS=1 to reduce parallel computing and potential crash on Jetson

Potential problems you could face:\

1. Permission denied, fix: MAX_JOBS=1 python setup.py install --user