From 74c02405452388bd5d0b3e9f96abb9c04e8a27d8 Mon Sep 17 00:00:00 2001 From: kaenafwi Date: Thu, 13 Aug 2026 00:53:12 +0000 Subject: [PATCH] Updated Dockerfiles based on build artifacts uploaded to the Neuron DLC Builder prod account --- jax/training/0.9.0/Dockerfile.neuronx | 3 ++- pytorch/inference/2.9.0/Dockerfile.neuronx | 3 ++- vllm/inference/0.16.0/Dockerfile.neuronx | 3 ++- vllm/inference/0.21.0.1.0.0/Dockerfile.neuronx | 3 ++- 4 files changed, 8 insertions(+), 4 deletions(-) diff --git a/jax/training/0.9.0/Dockerfile.neuronx b/jax/training/0.9.0/Dockerfile.neuronx index 10c7b21..235e8c3 100644 --- a/jax/training/0.9.0/Dockerfile.neuronx +++ b/jax/training/0.9.0/Dockerfile.neuronx @@ -26,6 +26,7 @@ ENV LD_LIBRARY_PATH="${LD_LIBRARY_PATH}:/opt/amazon/efa/lib64" ENV LD_LIBRARY_PATH="${LD_LIBRARY_PATH}:/opt/amazon/openmpi/lib64" ENV LD_LIBRARY_PATH="${LD_LIBRARY_PATH}:/usr/local/lib" ENV PATH="/opt/aws/neuron/bin:${PATH}" +ENV PATH="${PATH}:/opt/amazon/efa/bin" RUN apt-get update \ && apt-get upgrade -y \ @@ -173,7 +174,7 @@ RUN mkdir -p /etc/apt/keyrings \ # Neuron SDK components version numbers ARG NEURONX_RUNTIME_LIB_VERSION=2.33.10.0-3dcef56f0 ARG NEURONX_COLLECTIVES_LIB_VERSION=2.33.10.0-068180c7a -ARG NEURONX_TOOLS_VERSION=2.31.13.0-a9e473f33 +ARG NEURONX_TOOLS_VERSION=2.31.15.0-5c7949a6a ARG NEURONX_CC_VERSION=2.26.6360.0+6f180f47 ARG NEURONX_JAX_TRAINING_VERSION=0.10.0.1.0.10466+f1172f09 diff --git a/pytorch/inference/2.9.0/Dockerfile.neuronx b/pytorch/inference/2.9.0/Dockerfile.neuronx index bacc335..ca43200 100644 --- a/pytorch/inference/2.9.0/Dockerfile.neuronx +++ b/pytorch/inference/2.9.0/Dockerfile.neuronx @@ -18,6 +18,7 @@ ARG PYPI_SIMPLE_URL="https://pypi.org/simple/" ENV LANG=C.UTF-8 ENV LD_LIBRARY_PATH=/opt/aws/neuron/lib:/lib/x86_64-linux-gnu:/opt/conda/lib/:$LD_LIBRARY_PATH ENV PATH=/opt/conda/bin:/opt/aws/neuron/bin:$PATH +ENV PATH="${PATH}:/opt/amazon/efa/bin" RUN apt-get update \ && apt-get upgrade -y \ @@ -181,7 +182,7 @@ RUN mkdir -p /etc/apt/keyrings \ # Neuron SDK components version numbers ARG NEURONX_COLLECTIVES_LIB_VERSION=2.33.10.0-068180c7a ARG NEURONX_RUNTIME_LIB_VERSION=2.33.10.0-3dcef56f0 -ARG NEURONX_TOOLS_VERSION=2.31.13.0-a9e473f33 +ARG NEURONX_TOOLS_VERSION=2.31.15.0-5c7949a6a ARG NEURONX_CC_VERSION=2.26.6360.0+6f180f47 ARG NEURONX_FRAMEWORK_VERSION=2.9.0.2.15.32035+de43f57c diff --git a/vllm/inference/0.16.0/Dockerfile.neuronx b/vllm/inference/0.16.0/Dockerfile.neuronx index 8319936..1b23ac7 100644 --- a/vllm/inference/0.16.0/Dockerfile.neuronx +++ b/vllm/inference/0.16.0/Dockerfile.neuronx @@ -17,6 +17,7 @@ ARG PYPI_SIMPLE_URL="https://pypi.org/simple/" ENV LANG=C.UTF-8 ENV LD_LIBRARY_PATH=/opt/aws/neuron/lib:/lib/x86_64-linux-gnu:/opt/conda/lib/:$LD_LIBRARY_PATH ENV PATH=/opt/conda/bin:/opt/aws/neuron/bin:$PATH +ENV PATH="${PATH}:/opt/amazon/efa/bin" RUN apt-get update \ && apt-get upgrade -y \ @@ -159,7 +160,7 @@ RUN mkdir -p /etc/apt/keyrings \ # Neuron SDK components version numbers ARG NEURONX_COLLECTIVES_LIB_VERSION=2.33.10.0-068180c7a ARG NEURONX_RUNTIME_LIB_VERSION=2.33.10.0-3dcef56f0 -ARG NEURONX_TOOLS_VERSION=2.31.13.0-a9e473f33 +ARG NEURONX_TOOLS_VERSION=2.31.15.0-5c7949a6a ARG NEURONX_CC_VERSION=2.26.6360.0+6f180f47 ARG NEURONX_FRAMEWORK_VERSION=2.9.0.2.15.32035+de43f57c diff --git a/vllm/inference/0.21.0.1.0.0/Dockerfile.neuronx b/vllm/inference/0.21.0.1.0.0/Dockerfile.neuronx index cf69481..2bd67b7 100644 --- a/vllm/inference/0.21.0.1.0.0/Dockerfile.neuronx +++ b/vllm/inference/0.21.0.1.0.0/Dockerfile.neuronx @@ -17,6 +17,7 @@ ARG PYPI_SIMPLE_URL="https://pypi.org/simple/" ENV LANG=C.UTF-8 ENV LD_LIBRARY_PATH=/opt/aws/neuron/lib:/lib/x86_64-linux-gnu:/opt/conda/lib/:$LD_LIBRARY_PATH ENV PATH=/opt/conda/bin:/opt/aws/neuron/bin:$PATH +ENV PATH="${PATH}:/opt/amazon/efa/bin" RUN apt-get update \ && apt-get upgrade -y \ @@ -159,7 +160,7 @@ RUN mkdir -p /etc/apt/keyrings \ # Neuron SDK components version numbers ARG NEURONX_COLLECTIVES_LIB_VERSION=2.33.10.0-068180c7a ARG NEURONX_RUNTIME_LIB_VERSION=2.33.10.0-3dcef56f0 -ARG NEURONX_TOOLS_VERSION=2.31.13.0-a9e473f33 +ARG NEURONX_TOOLS_VERSION=2.31.15.0-5c7949a6a ARG NEURONX_CC_VERSION=2.26.6360.0+6f180f47 ARG NKI_VERSION=0.5.0+28631259367.ga768afa6