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 []