Introducing AWS Deep Learning AMI Neuron PyTorch — AWS Neuron Documentation (original) (raw)

This document is relevant for: Inf2, Trn1, Trn2

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, Trn2