Pytorch Neuron Setup — AWS Neuron Documentation (original) (raw)

This document is relevant for: Inf1, Inf2, Trn1, Trn2, Trn3

Pytorch Neuron Setup#

PyTorch Neuron (torch-neuronx) Setup for Inf2, Trn1, and Trn2 Instances

PyTorch Neuron (torch-neuron) Setup for Inf1 Instances

This document is relevant for: Inf1, Inf2, Trn1, Trn2, Trn3