This document is relevant for: Inf2, Trn1, Trn1n

Tensorflow Neuron (tensorflow-neuronx) Setup#

Note

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

tensorflow-neuronx setup on Ubuntu 20#

Ubuntu 20 (Ubuntu20 AMI)
Ubuntu 20 (DLAMI Base AMI)
Ubuntu 20 (DLAMI Tensorflow AMI)

tensorflow-neuronx setup on Ubuntu 22#

Ubuntu 22 (Neuron Multi-Framework DLAMI)
Ubuntu 22 (Ubuntu22 AMI)

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

Amazon Linux 2 (DLAMI Base AMI)
Amazon Linux 2 (DLAMI Tensorflow AMI)

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

Amazon Linux 2023 (Amazon Linux 2023 AMI)

This document is relevant for: Inf2, Trn1, Trn1n