```bash
python -m pip install --upgrade-strategy eager optimum[neuronx]
```  
<Tip>  
We provide pre-built [Hugging Face Neuron Deep Learning AMI](https://aws.amazon.com/marketplace/pp/prodview-gr3e6yiscria2) (DLAMI) and Optimum Neuron containers for Amazon SageMaker. It's recommended to correctly set up your environment.  
</Tip>