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 # =========================================================