.. _setup-torch-neuronx:

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

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

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

``torch-neuronx`` setup on Ubuntu 20 
------------------------------------

.. card:: Ubuntu 20 (Ubuntu20 AMI)
        :link: setup-torch-neuronx-ubuntu20
        :link-type: ref
        :class-body: sphinx-design-class-title-small

.. card:: Ubuntu 20 (DLAMI Base AMI)
        :link: setup-torch-neuronx-ubuntu20-base-dlami
        :link-type: ref
        :class-body: sphinx-design-class-title-small

.. card:: Ubuntu 20 (DLAMI PyTorch Neuron)
        :link: setup-torch-neuronx-ubuntu20-dlami-pytorch
        :link-type: ref
        :class-body: sphinx-design-class-title-small


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

.. 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 2 (AL2)
-----------------------------------------------


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

.. card:: Amazon Linux 2  (DLAMI Base AMI)
        :link: setup-torch-neuronx-al2-base-dlami
        :link-type: ref
        :class-body: sphinx-design-class-title-small

.. card:: Amazon Linux 2 (DLAMI PyTorch Neuron)
        :link: setup-torch-neuronx-al2-dlami-pytorch
        :link-type: ref
        :class-body: sphinx-design-class-title-small

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