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>
  • Loading branch information
mgoin and DarkLight1337 authored Oct 9, 2024
1 parent cdca899 commit d5fbb87
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 d5fbb87

Please sign in to comment.