.. _setup-torch-neuronx:

PyTorch Neuron (``torch-neuronx``) Setup 
========================================

.. note::
   This Setup guide is relevant for ``Inf2`` & ``Trn1`` / ``Trn1n`` / ``Trn2`` instances.

.. contents:: Table of contents
   :local:
   :depth: 2

``torch-neuronx`` setup on Ubuntu 24
------------------------------------

.. toctree::
   :maxdepth: 1
   :hidden:

   PyTorch NeuronX on Multi-Framework DLAMI (Ubuntu 24) </setup/neuron-setup/multiframework/multi-framework-ubuntu24-neuron-dlami>
   PyTorch NeuronX on Ubuntu 24 </setup/neuron-setup/pytorch/neuronx/ubuntu/torch-neuronx-ubuntu24>

.. card:: Ubuntu 24 (Neuron Multi-Framework DLAMI)
        :link: setup-ubuntu24-multi-framework-dlami
        :link-type: ref
        :class-body: sphinx-design-class-title-small

.. card:: Ubuntu 24 (Ubuntu24 AMI)
        :link: setup-torch-neuronx-ubuntu24
        :link-type: ref
        :class-body: sphinx-design-class-title-small

``torch-neuronx`` setup on Ubuntu 22
------------------------------------

.. toctree::
   :maxdepth: 1
   :hidden:

   PyTorch NeuronX on Multi-Framework DLAMI (Ubuntu 22) </setup/neuron-setup/multiframework/multi-framework-ubuntu22-neuron-dlami>
   PyTorch NeuronX on Ubuntu 22 </setup/neuron-setup/pytorch/neuronx/ubuntu/torch-neuronx-ubuntu22>

.. card:: Ubuntu 22 (Neuron Multi-Framework DLAMI)
        :link: setup-ubuntu22-multi-framework-dlami
        :link-type: ref
        :class-body: sphinx-design-class-title-small

.. card:: Ubuntu 22 (Ubuntu22 AMI)
        :link: setup-torch-neuronx-ubuntu22
        :link-type: ref
        :class-body: sphinx-design-class-title-small


``torch-neuronx`` setup on Amazon Linux 2023 (AL2023)
-----------------------------------------------------

.. toctree::
   :maxdepth: 1
   :hidden:

   PyTorch NeuronX on Amazon Linux 2023 </setup/neuron-setup/pytorch/neuronx/amazon-linux/torch-neuronx-al2023>

.. card:: Amazon Linux 2023 (Amazon Linux 2023 AMI)
        :link: setup-torch-neuronx-al2023
        :link-type: ref
        :class-body: sphinx-design-class-title-small

``torch-neuronx`` setup on Rocky Linux 9
----------------------------------------

.. toctree::
   :maxdepth: 1
   :hidden:

   PyTorch NeuronX on Rocky Linux 9 </setup/setup-rocky-linux-9>

.. card:: Rocky Linux 9 (Rocky Linux 9 AMI)
        :link: setup-rocky-linux-9
        :link-type: ref
        :class-body: sphinx-design-class-title-small