From 399948a725ae0f6125c793473f56c5cc8f4eeae0 Mon Sep 17 00:00:00 2001 From: eugr Date: Mon, 24 Nov 2025 17:02:04 -0800 Subject: [PATCH] Added missing modules for flashinfer --- Dockerfile | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/Dockerfile b/Dockerfile index 18da715..2b54350 100644 --- a/Dockerfile +++ b/Dockerfile @@ -55,7 +55,8 @@ RUN pip install xgrammar triton termplotlib # Note: Using the same index URLs as provided in your script RUN 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-jit-cache --index-url https://flashinfer.ai/whl/cu130 && \ + pip install apache-tvm-ffi nvidia-cudnn-frontend nvidia-cutlass-dsl nvidia-ml-py tabulate # Install fast safetensors to improve loading speeds RUN pip install fastsafetensors