.. _torch-setup:


.. meta::
   :description: Pytorch Neuron Setup - AWS Neuron SDK documentation
   :keywords: AWS Neuron, PyTorch, setup
   :date-modified: 2026-03-13


Pytorch Neuron Setup
====================

Install and configure PyTorch for use with AWS Neuron on Trainium and Inferentia instances.

.. grid:: 2
    :gutter: 2

    .. grid-item-card:: PyTorch Neuron (``torch-neuronx``) Setup for Inf2, Trn1, and Trn2 Instances
        :link: setup-torch-neuronx
        :link-type: ref
        :class-header: sd-bg-primary sd-text-white

        Install and configure PyTorch NeuronX for Inf2, Trn1, and Trn2 instances.

    .. grid-item-card:: PyTorch Neuron (``torch-neuron``) Setup for Inf1 Instances (Archived)
        :link: /archive/torch-neuron/setup/pytorch-install
        :link-type: doc
        :class-header: sd-bg-primary sd-text-white

        Legacy setup guide for PyTorch Neuron on Inf1 instances. This package has been archived.