Skip to content

Updated Dockerfiles #197

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
aws-dclusin merged 2 commits into from
Jul 8, 2026
Merged

Updated Dockerfiles #197

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
1673e41
Updated Dockerfiles based on build artifacts uploaded to the Neuron D…
Jul 8, 2026
493ebd5
Update vllm-neuron branch to release-0.5.2
aws-dclusin Jul 8, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
16 changes: 8 additions & 8 deletions jax/training/0.9.0/Dockerfile.neuronx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@
ENV PYTHONIOENCODING=UTF-8
ENV LANG=C.UTF-8
ENV LC_ALL=C.UTF-8
ENV LD_LIBRARY_PATH="${LD_LIBRARY_PATH}:/opt/aws/neuron/lib"

Check warning on line 23 in jax/training/0.9.0/Dockerfile.neuronx

View workflow job for this annotation

GitHub Actions / build (0.9.0, jax/training/0.9.0)

Variables should be defined before their use 

UndefinedVar: Usage of undefined variable '$LD_LIBRARY_PATH'
More info: https://docs.docker.com/go/dockerfile/rule/undefined-var/
ENV LD_LIBRARY_PATH="${LD_LIBRARY_PATH}:/opt/amazon/efa/lib"
ENV LD_LIBRARY_PATH="${LD_LIBRARY_PATH}:/opt/amazon/efa/lib64"
ENV LD_LIBRARY_PATH="${LD_LIBRARY_PATH}:/opt/amazon/openmpi/lib64"
Expand Down Expand Up @@ -162,23 +162,23 @@

# Setting up APT and PIP repo for neuron artifacts
ARG NEURON_APT_REPO=apt.repos.neuron.amazonaws.com
ARG NEURON_APT_REPO_KEY

Check warning on line 165 in jax/training/0.9.0/Dockerfile.neuronx

View workflow job for this annotation

GitHub Actions / build (0.9.0, jax/training/0.9.0)

Sensitive data should not be used in the ARG or ENV commands 

SecretsUsedInArgOrEnv: Do not use ARG or ENV instructions for sensitive data (ARG "NEURON_APT_REPO_KEY")
More info: https://docs.docker.com/go/dockerfile/rule/secrets-used-in-arg-or-env/
ARG NEURON_PIP_REPO=pip.repos.neuron.amazonaws.com
ARG NEURON_PIP_REPO_KEY

Check warning on line 167 in jax/training/0.9.0/Dockerfile.neuronx

View workflow job for this annotation

GitHub Actions / build (0.9.0, jax/training/0.9.0)

Sensitive data should not be used in the ARG or ENV commands 

SecretsUsedInArgOrEnv: Do not use ARG or ENV instructions for sensitive data (ARG "NEURON_PIP_REPO_KEY")
More info: https://docs.docker.com/go/dockerfile/rule/secrets-used-in-arg-or-env/
RUN mkdir -p /etc/apt/keyrings \
&& APT_REPO_PREFIX=$([ -n "${NEURON_APT_REPO_KEY}" ] && echo "${NEURON_APT_REPO_KEY}@" || echo "") \
&& echo "deb [signed-by=/etc/apt/keyrings/neuron.gpg] https://${APT_REPO_PREFIX}${NEURON_APT_REPO} jammy main" > /etc/apt/sources.list.d/neuron.list \
&& curl $([ -n "${NEURON_APT_REPO_KEY}" ] && echo "-u ${NEURON_APT_REPO_KEY}") --retry 3 --retry-delay 1 --retry-all-errors -fSL "https://${NEURON_APT_REPO}/GPG-PUB-KEY-AMAZON-AWS-NEURON.PUB" | gpg --dearmor > /etc/apt/keyrings/neuron.gpg

# Neuron SDK components version numbers
ARG NEURONX_RUNTIME_LIB_VERSION=2.32.31.0-0234f5ed2
ARG NEURONX_COLLECTIVES_LIB_VERSION=2.32.28.0-452cba8de
ARG NEURONX_TOOLS_VERSION=2.30.10.0-39feaa591

ARG NEURONX_CC_VERSION=2.25.3371.0+f524f7f8
ARG NEURONX_JAX_TRAINING_VERSION=0.10.0.1.0.9913+41e8ced2
ARG NKI_VERSION=0.4.0+25940409122.gd30719f9
ARG NEURON_AGENTIC_DEVELOPMENT_VERSION=1.1
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_CC_VERSION=2.26.6360.0+6f180f47
ARG NEURONX_JAX_TRAINING_VERSION=0.10.0.1.0.10466+f1172f09
ARG NKI_VERSION=0.5.0+28631259367.ga768afa6
ARG NEURON_AGENTIC_DEVELOPMENT_VERSION=1.2

FROM base AS repo

Expand Down
20 changes: 10 additions & 10 deletions pytorch/inference/2.9.0/Dockerfile.neuronx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@

# See http://bugs.python.org/issue19846
ENV LANG=C.UTF-8
ENV LD_LIBRARY_PATH=/opt/aws/neuron/lib:/lib/x86_64-linux-gnu:/opt/conda/lib/:$LD_LIBRARY_PATH

Check warning on line 19 in pytorch/inference/2.9.0/Dockerfile.neuronx

View workflow job for this annotation

GitHub Actions / build (2.9.0, pytorch/inference/2.9.0)

Variables should be defined before their use 

UndefinedVar: Usage of undefined variable '$LD_LIBRARY_PATH'
More info: https://docs.docker.com/go/dockerfile/rule/undefined-var/
ENV PATH=/opt/conda/bin:/opt/aws/neuron/bin:$PATH

RUN apt-get update \
Expand Down Expand Up @@ -170,25 +170,25 @@

# Setting up APT and PIP repo for neuron artifacts
ARG NEURON_APT_REPO=apt.repos.neuron.amazonaws.com
ARG NEURON_APT_REPO_KEY

Check warning on line 173 in pytorch/inference/2.9.0/Dockerfile.neuronx

View workflow job for this annotation

GitHub Actions / build (2.9.0, pytorch/inference/2.9.0)

Sensitive data should not be used in the ARG or ENV commands 

SecretsUsedInArgOrEnv: Do not use ARG or ENV instructions for sensitive data (ARG "NEURON_APT_REPO_KEY")
More info: https://docs.docker.com/go/dockerfile/rule/secrets-used-in-arg-or-env/
ARG NEURON_PIP_REPO=pip.repos.neuron.amazonaws.com
ARG NEURON_PIP_REPO_KEY

Check warning on line 175 in pytorch/inference/2.9.0/Dockerfile.neuronx

View workflow job for this annotation

GitHub Actions / build (2.9.0, pytorch/inference/2.9.0)

Sensitive data should not be used in the ARG or ENV commands 

SecretsUsedInArgOrEnv: Do not use ARG or ENV instructions for sensitive data (ARG "NEURON_PIP_REPO_KEY")
More info: https://docs.docker.com/go/dockerfile/rule/secrets-used-in-arg-or-env/
RUN mkdir -p /etc/apt/keyrings \
&& APT_REPO_PREFIX=$([ -n "${NEURON_APT_REPO_KEY}" ] && echo "${NEURON_APT_REPO_KEY}@" || echo "") \
&& echo "deb [signed-by=/etc/apt/keyrings/neuron.gpg] https://${APT_REPO_PREFIX}${NEURON_APT_REPO} jammy main" > /etc/apt/sources.list.d/neuron.list \
&& curl $([ -n "${NEURON_APT_REPO_KEY}" ] && echo "-u ${NEURON_APT_REPO_KEY}") --retry 3 --retry-delay 1 --retry-all-errors -fSL "https://${NEURON_APT_REPO}/GPG-PUB-KEY-AMAZON-AWS-NEURON.PUB" | gpg --dearmor > /etc/apt/keyrings/neuron.gpg

# Neuron SDK components version numbers
ARG NEURONX_COLLECTIVES_LIB_VERSION=2.32.28.0-452cba8de
ARG NEURONX_RUNTIME_LIB_VERSION=2.32.31.0-0234f5ed2
ARG NEURONX_TOOLS_VERSION=2.30.10.0-39feaa591

ARG NEURONX_CC_VERSION=2.25.3371.0+f524f7f8
ARG NEURONX_FRAMEWORK_VERSION=2.9.0.2.14.27725+e2ff0410
ARG NEURONX_DISTRIBUTED_VERSION=0.19.28093+fc70b593
ARG NEURONX_DISTRIBUTED_INFERENCE_VERSION=0.10.17970+8548ba25
ARG NKI_VERSION=0.4.0+25940409122.gd30719f9
ARG NEURON_AGENTIC_DEVELOPMENT_VERSION=1.1
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_CC_VERSION=2.26.6360.0+6f180f47
ARG NEURONX_FRAMEWORK_VERSION=2.9.0.2.15.32035+de43f57c
ARG NEURONX_DISTRIBUTED_VERSION=0.19.28492+435aae2b
ARG NEURONX_DISTRIBUTED_INFERENCE_VERSION=0.10.18399+ed62453e
ARG NKI_VERSION=0.5.0+28631259367.ga768afa6
ARG NEURON_AGENTIC_DEVELOPMENT_VERSION=1.2

FROM base AS repo

Expand Down
24 changes: 12 additions & 12 deletions vllm/inference/0.16.0/Dockerfile.neuronx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@

# See http://bugs.python.org/issue19846
ENV LANG=C.UTF-8
ENV LD_LIBRARY_PATH=/opt/aws/neuron/lib:/lib/x86_64-linux-gnu:/opt/conda/lib/:$LD_LIBRARY_PATH

Check warning on line 18 in vllm/inference/0.16.0/Dockerfile.neuronx

View workflow job for this annotation

GitHub Actions / build (0.16.0, vllm/inference/0.16.0)

Variables should be defined before their use 

UndefinedVar: Usage of undefined variable '$LD_LIBRARY_PATH'
More info: https://docs.docker.com/go/dockerfile/rule/undefined-var/
ENV PATH=/opt/conda/bin:/opt/aws/neuron/bin:$PATH

RUN apt-get update \
Expand Down Expand Up @@ -148,29 +148,29 @@

# Setting up APT and PIP repo for neuron artifacts
ARG NEURON_APT_REPO=apt.repos.neuron.amazonaws.com
ARG NEURON_APT_REPO_KEY

Check warning on line 151 in vllm/inference/0.16.0/Dockerfile.neuronx

View workflow job for this annotation

GitHub Actions / build (0.16.0, vllm/inference/0.16.0)

Sensitive data should not be used in the ARG or ENV commands 

SecretsUsedInArgOrEnv: Do not use ARG or ENV instructions for sensitive data (ARG "NEURON_APT_REPO_KEY")
More info: https://docs.docker.com/go/dockerfile/rule/secrets-used-in-arg-or-env/
ARG NEURON_PIP_REPO=pip.repos.neuron.amazonaws.com
ARG NEURON_PIP_REPO_KEY

Check warning on line 153 in vllm/inference/0.16.0/Dockerfile.neuronx

View workflow job for this annotation

GitHub Actions / build (0.16.0, vllm/inference/0.16.0)

Sensitive data should not be used in the ARG or ENV commands 

SecretsUsedInArgOrEnv: Do not use ARG or ENV instructions for sensitive data (ARG "NEURON_PIP_REPO_KEY")
More info: https://docs.docker.com/go/dockerfile/rule/secrets-used-in-arg-or-env/
RUN mkdir -p /etc/apt/keyrings \
&& APT_REPO_PREFIX=$([ -n "${NEURON_APT_REPO_KEY}" ] && echo "${NEURON_APT_REPO_KEY}@" || echo "") \
&& echo "deb [signed-by=/etc/apt/keyrings/neuron.gpg] https://${APT_REPO_PREFIX}${NEURON_APT_REPO} jammy main" > /etc/apt/sources.list.d/neuron.list \
&& curl $([ -n "${NEURON_APT_REPO_KEY}" ] && echo "-u ${NEURON_APT_REPO_KEY}") --retry 3 --retry-delay 1 --retry-all-errors -fSL "https://${NEURON_APT_REPO}/GPG-PUB-KEY-AMAZON-AWS-NEURON.PUB" | gpg --dearmor > /etc/apt/keyrings/neuron.gpg

# Neuron SDK components version numbers
ARG NEURONX_COLLECTIVES_LIB_VERSION=2.32.28.0-452cba8de
ARG NEURONX_RUNTIME_LIB_VERSION=2.32.31.0-0234f5ed2
ARG NEURONX_TOOLS_VERSION=2.30.10.0-39feaa591
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_CC_VERSION=2.25.3371.0+f524f7f8
ARG NEURONX_FRAMEWORK_VERSION=2.9.0.2.14.27725+e2ff0410
ARG NEURONX_DISTRIBUTED_VERSION=0.19.28093+fc70b593
ARG NEURONX_DISTRIBUTED_INFERENCE_VERSION=0.10.17970+8548ba25
ARG NKI_VERSION=0.4.0+25940409122.gd30719f9
ARG NEURON_AGENTIC_DEVELOPMENT_VERSION=1.1
ARG NEURONX_CC_VERSION=2.26.6360.0+6f180f47
ARG NEURONX_FRAMEWORK_VERSION=2.9.0.2.15.32035+de43f57c
ARG NEURONX_DISTRIBUTED_VERSION=0.19.28492+435aae2b
ARG NEURONX_DISTRIBUTED_INFERENCE_VERSION=0.10.18399+ed62453e
ARG NKI_VERSION=0.5.0+28631259367.ga768afa6
ARG NEURON_AGENTIC_DEVELOPMENT_VERSION=1.2

# GitHub repository and branch
ARG GITHUB_REPO=https://github.com/vllm-project/vllm-neuron.git
ARG GITHUB_REPO_BRANCH=release-0.5.1
ARG GITHUB_REPO_BRANCH=release-0.5.2

# Configure SSH access
RUN mkdir -p /root/.ssh \
Expand Down Expand Up @@ -205,7 +205,7 @@
neuronx_distributed_inference \
nki \
neuron_agentic_development \
-e /opt/vllm \
/opt/vllm \
&& rm -rf ~/.cache/pip/*

FROM base AS prod
Expand All @@ -231,7 +231,7 @@
neuronx_distributed_inference==$NEURONX_DISTRIBUTED_INFERENCE_VERSION \
nki==$NKI_VERSION \
neuron_agentic_development==$NEURON_AGENTIC_DEVELOPMENT_VERSION \
-e /opt/vllm \
/opt/vllm \
&& rm -rf ~/.cache/pip/*

FROM ${BUILD_STAGE} AS final
Expand Down
Loading