Skip to content

GitLab

Unverified Commit aebfcb26 authored by GhaziSyed's avatar GhaziSyed Committed by GitHub
Browse files

Dockerfile: Upgrade Cuda to 12.1 (#1609)

parent ab9e8488
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
Dockerfile
View file @ aebfcb26
FROM nvidia/cuda:11.8.0-devel-ubuntu22.04 AS dev FROM nvidia/cuda:12.1.0-devel-ubuntu22.04 AS dev
RUN apt-get update -y \ RUN apt-get update -y \
&& apt-get install -y python3-pip && apt-get install -y python3-pip
...@@ -41,7 +41,7 @@ COPY vllm vllm ...@@ -41,7 +41,7 @@ COPY vllm vllm
ENTRYPOINT ["python3", "-m", "pytest", "tests"] ENTRYPOINT ["python3", "-m", "pytest", "tests"]
# use CUDA base as CUDA runtime dependencies are already installed via pip # use CUDA base as CUDA runtime dependencies are already installed via pip
FROM nvidia/cuda:11.8.0-base-ubuntu22.04 AS vllm-base FROM nvidia/cuda:12.1.0-base-ubuntu22.04 AS vllm-base
# libnccl required for ray # libnccl required for ray
RUN apt-get update -y \ RUN apt-get update -y \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment