Pytorch 1.7.1 with Magma, available on Anaconda cloud

I built my own Pytorch because the official one isn’t compiled with Magma. Magma is necessary to run some software packages such as 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



Thanks for doing this and sharing.
We will share this with the user who also wants this.