.. _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 22
------------------------------------------

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

   TensorFlow NeuronX on Multi-Framework DLAMI (Ubuntu 22) </setup/neuron-setup/multiframework/multi-framework-ubuntu22-neuron-dlami>
   TensorFlow NeuronX on Ubuntu 22 </setup/neuron-setup/tensorflow/neuronx/ubuntu/tensorflow-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-tensorflow-neuronx-u22
        :link-type: ref
        :class-body: sphinx-design-class-title-small


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

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

   TensorFlow NeuronX on Amazon Linux 2023 </setup/neuron-setup/tensorflow/neuronx/amazon-linux/tensorflow-neuronx-al2023>

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