Pytorch 1.7.1 with Magma, available on Anaconda cloud

Robotics & Edge Computing Jetson & Embedded Systems Jetson AGX Xavier
1

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

David

3

Hi,

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

Thanks.