From cf0a7a5475dad1e559fa0392a1b5cdcd0312af77 Mon Sep 17 00:00:00 2001 From: tzhouam Date: Tue, 10 Mar 2026 08:05:28 +0000 Subject: [PATCH] Update Dockerfile.ci to install additional dependencies for flashinfer and numpy, and create a symlink for python3. This enhances the CI environment setup. Signed-off-by: tzhouam --- docker/Dockerfile.ci | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/docker/Dockerfile.ci b/docker/Dockerfile.ci index 51c03cb7e..3b90ac261 100644 --- a/docker/Dockerfile.ci +++ b/docker/Dockerfile.ci @@ -21,4 +21,13 @@ RUN uv pip install --system --upgrade --force-reinstall --prerelease=allow vllm RUN uv pip install --system --no-cache-dir ".[dev]" +RUN uv pip install --system --upgrade \ + "flashinfer-cubin==0.6.4" \ + "nvidia-cublas-cu12==12.9.1.4" \ + "numpy==2.2.6" +RUN uv pip install --system --upgrade \ + "flashinfer-jit-cache==0.6.4" \ + --index-url https://flashinfer.ai/whl/cu129 +RUN ln -sf /usr/bin/python3 /usr/bin/python + ENTRYPOINT []