From 8dcf1061f13fd8331bbfa8c24c34363434783e7a Mon Sep 17 00:00:00 2001 From: tzhouam Date: Tue, 3 Mar 2026 06:47:10 +0000 Subject: [PATCH] Update Dockerfile.ci to install additional dependencies including flashinfer-jit-cache, nvidia-cublas-cu12, and numpy, ensuring a more comprehensive environment setup for development. Signed-off-by: tzhouam --- docker/Dockerfile.ci | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/docker/Dockerfile.ci b/docker/Dockerfile.ci index db895e1bd..be8cd4a01 100644 --- a/docker/Dockerfile.ci +++ b/docker/Dockerfile.ci @@ -21,8 +21,11 @@ RUN apt-get update && \ RUN uv pip install --system --upgrade --force-reinstall --prerelease=allow vllm --extra-index-url https://wheels.vllm.ai/6521ccf2860eb72cf23a9fa9044bf9721fd9a7c6 RUN uv pip install --system --no-cache-dir ".[dev]" -RUN uv pip install --system --upgrade "flashinfer-cubin==0.6.4" - +RUN uv pip install --system --upgrade \ + "flashinfer-cubin==0.6.4" \ + "flashinfer-jit-cache==0.6.4" \ + "nvidia-cublas-cu12==12.9.1.4" \ + "numpy==2.2.6" RUN ln -sf /usr/bin/python3 /usr/bin/python ENTRYPOINT []