This document is relevant for: Inf2, Trn1, Trn2

If you already have a previous Neuron release installed, select the PyTorch version tab below to get the update commands for your environment.

PyTorch 2.8.0

Note

• Instructions in this page only apply to setting up Neuron components on Linux host running Ubuntu or Amazon Linux AMI.

• When launching a Trn1/Trn2/Trn3 instance, you must adjust your primary EBS volume size to a minimum of 512GB.

Note

PyTorch versions 2.7 and 2.8 are no longer supported on Neuron. If you are looking for setup instructions specific to PyTorch 2.7 and 2.8 on Amazon Linux 2023, Ubuntu 24.04, or Ubuntu 22.04, see the Neuron release 2.28.0 version of the setup docs.

PyTorch 2.7.0

Note

• Instructions in this page only apply to setting up Neuron components on Linux host running Ubuntu or Amazon Linux AMI.

• When launching a Trn1/Trn2/Trn3 instance, you must adjust your primary EBS volume size to a minimum of 512GB.

Note

PyTorch versions 2.7 and 2.8 are no longer supported on Neuron. If you are looking for setup instructions specific to PyTorch 2.7 and 2.8 on Amazon Linux 2023, Ubuntu 24.04, or Ubuntu 22.04, see the Neuron release 2.28.0 version of the setup docs.

This document is relevant for: Inf2, Trn1, Trn2