This document is relevant for: Inf2, Trn1, Trn2

Tensorflow Neuron (tensorflow-neuronx) Setup#

Note

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

tensorflow-neuronx setup on Ubuntu 22#

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

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

Amazon Linux 2023 (Amazon Linux 2023 AMI)

This document is relevant for: Inf2, Trn1, Trn2