This document is relevant for: Inf2
, Trn1
, Trn2
PyTorch Neuron (torch-neuronx
) Setup#
Note
This Setup guide is relevant for Inf2
& Trn1
/ Trn1n
/ Trn2
instances.
torch-neuronx
setup on Ubuntu 20#
torch-neuronx
setup on Ubuntu 22#
torch-neuronx
setup on Amazon Linux 2023 (AL2023)#
torch-neuronx
setup on Rocky Linux 9#
This document is relevant for: Inf2
, Trn1
, Trn2