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 <tlleungac@connect.ust.hk>
Signed-off-by: Alvant <alvasian@yandex.ru>
  • Loading branch information
2 people authored and Alvant committed Oct 26, 2024
1 parent 63e38f1 commit 7140021
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 7140021

Please sign in to comment.