From f0624a93cedc06ce0ddcfa4877aa77fed0d3b61f Mon Sep 17 00:00:00 2001 From: kaenafwi Date: Sat, 2 May 2026 05:23:35 +0000 Subject: [PATCH] Updated Dockerfiles based on build artifacts uploaded to the Neuron DLC Builder prod account --- jax/training/0.7/Dockerfile.neuronx | 4 ++-- pytorch/inference/2.9.0/Dockerfile.neuronx | 6 +++--- pytorch/training/2.9.0/Dockerfile.neuronx | 6 +++--- vllm/inference/0.16.0/Dockerfile.neuronx | 8 ++++---- 4 files changed, 12 insertions(+), 12 deletions(-) diff --git a/jax/training/0.7/Dockerfile.neuronx b/jax/training/0.7/Dockerfile.neuronx index 21fa457..4c08562 100644 --- a/jax/training/0.7/Dockerfile.neuronx +++ b/jax/training/0.7/Dockerfile.neuronx @@ -173,9 +173,9 @@ RUN mkdir -p /etc/apt/keyrings \ # Neuron SDK components version numbers ARG NEURONX_RUNTIME_LIB_VERSION=2.31.24.0-0b044f4ce ARG NEURONX_COLLECTIVES_LIB_VERSION=2.31.24.0-1a31ba186 -ARG NEURONX_TOOLS_VERSION=2.29.18.0-d5fe7ba42 +ARG NEURONX_TOOLS_VERSION=2.29.22.0-b486b0ade -ARG NEURONX_CC_VERSION=2.24.5133.0+58f8de22 +ARG NEURONX_CC_VERSION=2.24.8799.0+6f62ff7c ARG NEURONX_JAX_TRAINING_VERSION=0.7.0.1.0.8181+1e892be0 ARG NKI_VERSION=0.3.0+23928721754.g18aa1271 diff --git a/pytorch/inference/2.9.0/Dockerfile.neuronx b/pytorch/inference/2.9.0/Dockerfile.neuronx index 12db83c..9cddadd 100644 --- a/pytorch/inference/2.9.0/Dockerfile.neuronx +++ b/pytorch/inference/2.9.0/Dockerfile.neuronx @@ -183,10 +183,10 @@ RUN mkdir -p /etc/apt/keyrings \ # Neuron SDK components version numbers ARG NEURONX_COLLECTIVES_LIB_VERSION=2.31.24.0-1a31ba186 ARG NEURONX_RUNTIME_LIB_VERSION=2.31.24.0-0b044f4ce -ARG NEURONX_TOOLS_VERSION=2.29.18.0-d5fe7ba42 +ARG NEURONX_TOOLS_VERSION=2.29.22.0-b486b0ade -ARG NEURONX_CC_VERSION=2.24.5133.0+58f8de22 -ARG NEURONX_FRAMEWORK_VERSION=2.9.0.2.13.24727+8e870898 +ARG NEURONX_CC_VERSION=2.24.8799.0+6f62ff7c +ARG NEURONX_FRAMEWORK_VERSION=2.9.0.2.13.26312+8e870898 ARG NEURONX_DISTRIBUTED_VERSION=0.18.27753+1cafd54f ARG NEURONX_DISTRIBUTED_INFERENCE_VERSION=0.9.17334+ced6ae4e ARG NKI_VERSION=0.3.0+23928721754.g18aa1271 diff --git a/pytorch/training/2.9.0/Dockerfile.neuronx b/pytorch/training/2.9.0/Dockerfile.neuronx index 6a64abe..ee0f035 100644 --- a/pytorch/training/2.9.0/Dockerfile.neuronx +++ b/pytorch/training/2.9.0/Dockerfile.neuronx @@ -209,10 +209,10 @@ RUN mkdir -p /etc/apt/keyrings \ # Neuron SDK components ARG NEURONX_COLLECTIVES_LIB_VERSION=2.31.24.0-1a31ba186 ARG NEURONX_RUNTIME_LIB_VERSION=2.31.24.0-0b044f4ce -ARG NEURONX_TOOLS_VERSION=2.29.18.0-d5fe7ba42 +ARG NEURONX_TOOLS_VERSION=2.29.22.0-b486b0ade -ARG NEURONX_FRAMEWORK_VERSION=2.9.0.2.13.24727+8e870898 -ARG NEURONX_CC_VERSION=2.24.5133.0+58f8de22 +ARG NEURONX_FRAMEWORK_VERSION=2.9.0.2.13.26312+8e870898 +ARG NEURONX_CC_VERSION=2.24.8799.0+6f62ff7c ARG NEURONX_DISTRIBUTED_VERSION=0.18.27753+1cafd54f ARG NEURONX_DISTRIBUTED_TRAINING_VERSION=1.7.0 ARG NKI_VERSION=0.3.0+23928721754.g18aa1271 diff --git a/vllm/inference/0.16.0/Dockerfile.neuronx b/vllm/inference/0.16.0/Dockerfile.neuronx index 3edd8f7..bc60407 100644 --- a/vllm/inference/0.16.0/Dockerfile.neuronx +++ b/vllm/inference/0.16.0/Dockerfile.neuronx @@ -109,7 +109,7 @@ COPY --chmod=755 vllm_entrypoint.py neuron-monitor.sh deep_learning_container.py ### Mount Point ### # When launching the container, mount the code directory to /workspace ARG APP_MOUNT=/workspace -VOLUME [ ${APP_MOUNT} ] +VOLUME ${APP_MOUNT} WORKDIR ${APP_MOUNT}/vllm # Install AWS CLI @@ -161,10 +161,10 @@ RUN mkdir -p /etc/apt/keyrings \ # Neuron SDK components version numbers ARG NEURONX_COLLECTIVES_LIB_VERSION=2.31.24.0-1a31ba186 ARG NEURONX_RUNTIME_LIB_VERSION=2.31.24.0-0b044f4ce -ARG NEURONX_TOOLS_VERSION=2.29.18.0-d5fe7ba42 +ARG NEURONX_TOOLS_VERSION=2.29.22.0-b486b0ade -ARG NEURONX_CC_VERSION=2.24.5133.0+58f8de22 -ARG NEURONX_FRAMEWORK_VERSION=2.9.0.2.13.24727+8e870898 +ARG NEURONX_CC_VERSION=2.24.8799.0+6f62ff7c +ARG NEURONX_FRAMEWORK_VERSION=2.9.0.2.13.26312+8e870898 ARG NEURONX_DISTRIBUTED_VERSION=0.18.27753+1cafd54f ARG NEURONX_DISTRIBUTED_INFERENCE_VERSION=0.9.17334+ced6ae4e ARG NKI_VERSION=0.3.0+23928721754.g18aa1271