Skip to content

Commit

Permalink
[CI/Build] Update Dockerfile install+deploy image to ubuntu 22.04 (vl…
Browse files Browse the repository at this point in the history
…lm-project#9130)

Co-authored-by: DarkLight1337 <[email protected]>
Signed-off-by: Maxime Fournioux <[email protected]>
  • Loading branch information
2 people authored and mfournioux committed Nov 20, 2024
1 parent 1f172c3 commit 5598983
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -144,7 +144,7 @@ RUN --mount=type=cache,target=/root/.cache/pip \
#################### DEV IMAGE ####################
#################### vLLM installation IMAGE ####################
# image with vLLM installed
FROM nvidia/cuda:${CUDA_VERSION}-base-ubuntu20.04 AS vllm-base
FROM nvidia/cuda:${CUDA_VERSION}-base-ubuntu22.04 AS vllm-base
ARG CUDA_VERSION=12.4.1
ARG PYTHON_VERSION=3.12
WORKDIR /vllm-workspace
Expand Down

0 comments on commit 5598983

Please sign in to comment.