This document is relevant for: Inf2, Trn1, Trn1n

Introducing AWS Deep Learning AMI Neuron PyTorch

We are happy to announce that Deep Learning AMI (DLAMI) with pre-installed PyTorch Neuron (torch-neuronx) is now available, for more information see:

• AWS Deep Learning AMI Neuron PyTorch 1.11 (Amazon Linux 2)

• AWS Deep Learning AMI Neuron PyTorch 1.11 (Ubuntu 20.04)

The Neuron Setup Guide will be updated soon to include the DLAMI PyTorch Neuron.

This document is relevant for: Inf2, Trn1, Trn1n