.. _setup-tensorflow-neuronx:

Tensorflow Neuron (``tensorflow-neuronx``) Setup
================================================

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

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


``tensorflow-neuronx`` setup on Ubuntu 20
------------------------------------------

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

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

.. card:: Ubuntu 20 (DLAMI Tensorflow AMI)
        :link: setup-tensorflow-neuronx-u20-dlami-tensorflow
        :link-type: ref
        :class-body: sphinx-design-class-title-small


``tensorflow-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-tensorflow-neuronx-u22
        :link-type: ref
        :class-body: sphinx-design-class-title-small

``tensorflow-neuronx`` setup on Amazon Linux 2 (AL2)
-----------------------------------------------------


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

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

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

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

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