Skip to content

Commit 1d01211

Browse files
Update BASE_IMAGE to 2.22 release of Neuron (#16218)
1 parent f94ab12 commit 1d01211

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

docker/Dockerfile.neuron

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
# default base image
22
# https://gallery.ecr.aws/neuron/pytorch-inference-neuronx
3-
ARG BASE_IMAGE="public.ecr.aws/neuron/pytorch-inference-neuronx:2.5.1-neuronx-py310-sdk2.21.0-ubuntu22.04"
3+
ARG BASE_IMAGE="public.ecr.aws/neuron/pytorch-inference-neuronx:2.5.1-neuronx-py310-sdk2.22.0-ubuntu22.04"
44

55
FROM $BASE_IMAGE
66

@@ -21,9 +21,9 @@ VOLUME [ ${APP_MOUNT} ]
2121
WORKDIR ${APP_MOUNT}/vllm
2222

2323
RUN python3 -m pip install --upgrade pip
24-
RUN python3 -m pip install --no-cache-dir fastapi ninja tokenizers pandas
25-
RUN python3 -m pip install sentencepiece transformers==4.45.2 -U
26-
RUN python3 -m pip install neuronx-cc==2.16.345.0 --extra-index-url=https://pip.repos.neuron.amazonaws.com -U
24+
RUN python3 -m pip install --no-cache-dir fastapi ninja tokenizers pandas tenacity
25+
RUN python3 -m pip install sentencepiece transformers==4.48.0 -U
26+
RUN python3 -m pip install neuronx-cc==2.17.194.0 --extra-index-url=https://pip.repos.neuron.amazonaws.com -U
2727
RUN python3 -m pip install pytest
2828

2929
# uninstall transformers-neuronx package explicitly to avoid version conflict

0 commit comments

Comments
 (0)