From 1 and 2 install the required version of PyTorch (check example.sh)
To build Torchvision from source:
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 2cd vision
export FORCE_CUDA=1
# Only for devices with CUDAMAX_JOBS=1 python setup.py install
# MAX_JOBS=1 to reduce parallel computing and potential crash on Jetson
Potential problems you could face:
- Permission denied, fix:
MAX_JOBS=1 python setup.py install --user