chore: update neuron dependencies (aws#663)

Co-authored-by: GitHub <[email protected]>

Author: 2 people

test/images/neuron-inference/Dockerfile

@@ -11,11 +11,11 @@ ENV PYTHONUNBUFFERED=1
 
 # Neuron SDK components version numbers
 # https://github.com/aws-neuron/deep-learning-containers/blob/main/docker/pytorch/inference/2.5.1/Dockerfile.neuronx
-ARG NEURONX_CC_VERSION=2.19.8089.0+8ab9f450
-ARG NEURONX_FRAMEWORK_VERSION=2.7.0.2.8.6734+ac864f72
-ARG NEURONX_COLLECTIVES_LIB_VERSION=2.26.43.0-47cc904ea
-ARG NEURONX_RUNTIME_LIB_VERSION=2.26.42.0-2ff3b5c7d
-ARG NEURONX_TOOLS_VERSION=2.24.54.0
+ARG NEURONX_CC_VERSION=2.20.9961.0+0acef03a
+ARG NEURONX_FRAMEWORK_VERSION=2.7.0.2.9.9357+08e1f40d
+ARG NEURONX_COLLECTIVES_LIB_VERSION=2.27.34.0-ec8cd5e8b
+ARG NEURONX_RUNTIME_LIB_VERSION=2.27.23.0-8deec4dbf
+ARG NEURONX_TOOLS_VERSION=2.25.145.0
 
 # Python
 ARG PYTHON=python3.10

test/images/neuron-training/Dockerfile

@@ -7,11 +7,11 @@ ARG DEBIAN_FRONTEND=noninteractive
 
 # Neuron SDK component versions (pin these precisely)
 # https://github.com/aws-neuron/deep-learning-containers/blob/main/docker/pytorch/training/2.5.1/Dockerfile.neuronx
-ARG NEURONX_CC_VERSION=2.19.8089.0+8ab9f450
-ARG NEURONX_FRAMEWORK_VERSION=2.7.0.2.8.6734+ac864f72
-ARG NEURONX_COLLECTIVES_LIB_VERSION=2.26.43.0-47cc904ea
-ARG NEURONX_RUNTIME_LIB_VERSION=2.26.42.0-2ff3b5c7d
-ARG NEURONX_TOOLS_VERSION=2.24.54.0
+ARG NEURONX_CC_VERSION=2.20.9961.0+0acef03a
+ARG NEURONX_FRAMEWORK_VERSION=2.7.0.2.9.9357+08e1f40d
+ARG NEURONX_COLLECTIVES_LIB_VERSION=2.27.34.0-ec8cd5e8b
+ARG NEURONX_RUNTIME_LIB_VERSION=2.27.23.0-8deec4dbf
+ARG NEURONX_TOOLS_VERSION=2.25.145.0
 
 # Python
 ARG PYTHON=python3.10

test/images/neuron/Dockerfile

@@ -2,12 +2,12 @@ FROM public.ecr.aws/docker/library/ubuntu:22.04
 
 # Neuron SDK components version numbers
 # https://github.com/aws-neuron/deep-learning-containers/blob/main/docker/pytorch/training/2.5.1/Dockerfile.neuronx
-ARG NEURONX_DISTRIBUTED_VERSION=0.13.14393+b8569585
-ARG NEURONX_CC_VERSION=2.19.8089.0+8ab9f450
-ARG NEURONX_FRAMEWORK_VERSION=2.7.0.2.8.6734+ac864f72
-ARG NEURONX_COLLECTIVES_LIB_VERSION=2.26.43.0-47cc904ea
-ARG NEURONX_RUNTIME_LIB_VERSION=2.26.42.0-2ff3b5c7d
-ARG NEURONX_TOOLS_VERSION=2.24.54.0
+ARG NEURONX_DISTRIBUTED_VERSION=0.14.18461+9ac233f2
+ARG NEURONX_CC_VERSION=2.20.9961.0+0acef03a
+ARG NEURONX_FRAMEWORK_VERSION=2.7.0.2.9.9357+08e1f40d
+ARG NEURONX_COLLECTIVES_LIB_VERSION=2.27.34.0-ec8cd5e8b
+ARG NEURONX_RUNTIME_LIB_VERSION=2.27.23.0-8deec4dbf
+ARG NEURONX_TOOLS_VERSION=2.25.145.0
 
 ARG PYTHON=python3.10
 ARG PYTHON_VERSION=3.10.12