From ac937b82e62ec55eab44bdf32714dc1b40b47efe Mon Sep 17 00:00:00 2001 From: Zhou Taichang Date: Tue, 17 Mar 2026 11:08:45 +0800 Subject: [PATCH] Refactor Dockerfile to install vLLM from source Signed-off-by: Zhou Taichang --- docker/Dockerfile.ci | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/docker/Dockerfile.ci b/docker/Dockerfile.ci index 2306dc48c..9ba24f2a5 100644 --- a/docker/Dockerfile.ci +++ b/docker/Dockerfile.ci @@ -15,11 +15,14 @@ RUN uv pip uninstall --system -y vllm || true # Install vLLM from source at the selected commit using precompiled artifacts ENV VLLM_PRECOMPILED_WHEEL_COMMIT=2754231ba3a72f41e62922d1552c33e8f3f6a9d1 -RUN uv pip install --system --force-reinstall --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 \ +RUN git clone https://github.com/vllm-project/vllm.git /workspace/vllm && \ + cd /workspace/vllm && \ + git checkout ${VLLM_PRECOMPILED_WHEEL_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 \ "flashinfer-cubin==0.6.6" \ "nvidia-cublas-cu12==12.9.1.4" \ "numpy==2.2.6"