update

Signed-off-by: yewentao256 <zhyanwentao@126.com>
2026-03-27 15:20:09 +00:00
parent 54dd55edb6
commit 388802e2fd
1 changed files with 1 additions and 0 deletions
@@ -594,6 +594,7 @@ ARG FLASHINFER_VERSION=0.6.6
 RUN --mount=type=cache,target=/root/.cache/uv \
    uv pip install --system flashinfer-jit-cache==${FLASHINFER_VERSION} \
        --extra-index-url https://flashinfer.ai/whl/cu$(echo $CUDA_VERSION | cut -d. -f1,2 | tr -d '.') \
+    && flashinfer download-cubin \
    && flashinfer show-config

 # ============================================================