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 <tzhouam@connect.ust.hk>

Signed-off-by: tzhouam <tzhouam@connect.ust.hk>
This commit is contained in:
tzhouam committed 2026-03-17 01:35:12 +00:00
1 parent 22a2bd3406
commit 83b05ce3e4
1 file changed
+8 -9
+8 -9
View File
@@ -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"