I built my own Pytorch because the official one isn’t compiled with Magma. Magma is necessary to run some software packages such as fast.ai. Also, there are some major bugs in v1.7.0 that break fastai (such as learn.summary); v1.7.1 fixes these bugs.
Step 1: download and install miniforge.
Step 2: conda install -c yinyang pytorch torchvision torchaudio
A docker image is also available. Install with docker pull yinyang/ml:jetson