diff --git a/docker/Dockerfile.ci b/docker/Dockerfile.ci index b9363e76a..a5ba60855 100644 --- a/docker/Dockerfile.ci +++ b/docker/Dockerfile.ci @@ -26,6 +26,8 @@ RUN uv pip uninstall --system nvidia-cublas-cu12 # Use bash -c so that $(python3 -c ...) is expanded inside the container. RUN uv pip install --python "$(python3 -c 'import sys; print(sys.executable)')" --no-cache-dir ".[dev]" +RUN pip install --system --upgrade --force-reinstall "pplx_kernels" + RUN /bin/bash -c 'FLASHINFER_CUDA_TAG="$(python3 -c "import torch; print((torch.version.cuda or \"12.4\").replace(\".\", \"\"))")" && \ uv pip install --system --upgrade --force-reinstall \ "flashinfer-python==0.6.3" \