feat: update dockerfile to 12.9.1 (#267)
Release / release (push) Waiting to run

* feat: update dockerfile to 12.9.1

* update readme on VLLM_NIGHTLY build arg
This commit is contained in:
chrisvela
2026-02-19 10:13:14 +01:00
committed by GitHub
parent d69cc021e8
commit b7c6d4f9a2
2 changed files with 18 additions and 3 deletions
+3 -3
View File
@@ -1,13 +1,13 @@
FROM nvidia/cuda:12.8.0-base-ubuntu22.04
FROM nvidia/cuda:12.9.1-base-ubuntu22.04
RUN apt-get update -y \
&& apt-get install -y python3-pip
RUN ldconfig /usr/local/cuda-12.8/compat/
RUN ldconfig /usr/local/cuda-12.9/compat/
# Install vLLM with FlashInfer - use CUDA 12.8 PyTorch wheels (compatible with vLLM 0.15.0)
RUN python3 -m pip install --upgrade pip && \
python3 -m pip install "vllm[flashinfer]==0.15.0" --extra-index-url https://download.pytorch.org/whl/cu128
python3 -m pip install "vllm[flashinfer]==0.15.0" --extra-index-url https://download.pytorch.org/whl/cu129