From 83b05ce3e450c07b5bf5daaf17c653ff766c3654 Mon Sep 17 00:00:00 2001 From: tzhouam Date: Tue, 17 Mar 2026 01:35:12 +0000 Subject: [PATCH] Refactor Dockerfile.ci to install vLLM using a precompiled wheel from a specified commit, replacing the previous source installation method. This update simplifies the installation process and ensures correct dependency management. Signed-off-by: Zhou Taichang Signed-off-by: tzhouam --- docker/Dockerfile.ci | 17 ++++++++--------- 1 file changed, 8 insertions(+), 9 deletions(-) diff --git a/docker/Dockerfile.ci b/docker/Dockerfile.ci index 1397c8f90..584034531 100644 --- a/docker/Dockerfile.ci +++ b/docker/Dockerfile.ci @@ -11,16 +11,15 @@ RUN apt-get update && \ apt-get clean && \ rm -rf /var/lib/apt/lists/* +RUN uv pip uninstall --system -y vllm || true + # Install vLLM from source at the selected commit using precompiled artifacts -ARG VLLM_COMMIT=2754231ba3a72f41e62922d1552c33e8f3f6a9d1 -RUN git clone https://github.com/vllm-project/vllm.git /workspace/vllm && \ - cd /workspace/vllm && \ - git checkout ${VLLM_COMMIT} && \ - (uv pip uninstall --system -y vllm || true) && \ - VLLM_USE_PRECOMPILED=1 uv pip install --system -e . && \ - cd ${APP_DIR} && \ - uv pip install --system ".[dev]" && \ - uv pip install --system --upgrade \ +ENV VLLM_PRECOMPILED_WHEEL_COMMIT=2754231ba3a72f41e62922d1552c33e8f3f6a9d1 +RUN uv pip install --system --prerelease=allow vllm --extra-index-url https://wheels.vllm.ai/${VLLM_PRECOMPILED_WHEEL_COMMIT} + +RUN uv pip install --system ".[dev]" + +RUN uv pip install --system --upgrade \ "flashinfer-cubin==0.6.6" \ "nvidia-cublas-cu12==12.9.1.4" \ "numpy==2.2.6"