This document is relevant for: Inf1, Inf2, Trn1, Trn1n

Setup Guide#

This section walks you through the various options to install and upgrade Neuron. You have to install Neuron on Trainium and Inferentia powered instances to enable deep-learning acceleration.

PyTorch Neuron (torch-neuronx) Setup for Inf2 & Trn1 / Trn1n Instances
PyTorch Neuron (torch-neuron) Setup for Inf1 Instances
Tensorflow Neuron (tensorflow-neuronx) Setup for Inf2 & Trn1 / Trn1n Instances
Tensorflow Neuron (tensorflow-neuron) Setup for Inf1 Instances
MxNet Neuron (mxnet-neuron) Setup for Inf1 Instances

This document is relevant for: Inf1, Inf2, Trn1, Trn1n