From fbb1bf73d58f616b3edd60c1f66248db5eeece90 Mon Sep 17 00:00:00 2001 From: Eugene Rakhmatulin Date: Sat, 20 Dec 2025 13:28:06 -0800 Subject: [PATCH] Switching to flashinfer 0.6.x pre-release wheels --- Dockerfile | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/Dockerfile b/Dockerfile index 2285d93..b257b93 100644 --- a/Dockerfile +++ b/Dockerfile @@ -69,9 +69,9 @@ RUN --mount=type=cache,id=pip-cache,target=/root/.cache/pip \ # Install FlashInfer packages RUN --mount=type=cache,id=pip-cache,target=/root/.cache/pip \ - pip install flashinfer-python --no-deps --index-url https://flashinfer.ai/whl && \ - pip install flashinfer-cubin --index-url https://flashinfer.ai/whl && \ - pip install flashinfer-jit-cache --index-url https://flashinfer.ai/whl/cu130 && \ + pip install flashinfer-python --no-deps --index-url https://flashinfer.ai/whl --pre && \ + pip install flashinfer-cubin --index-url https://flashinfer.ai/whl --pre && \ + pip install flashinfer-jit-cache --index-url https://flashinfer.ai/whl/cu130 --pre && \ pip install apache-tvm-ffi nvidia-cudnn-frontend nvidia-cutlass-dsl nvidia-ml-py tabulate # =========================================================