From 57d862d695034c878d26ab290c1a4a9def992267 Mon Sep 17 00:00:00 2001 From: hailin Date: Sat, 20 Sep 2025 10:33:21 +0800 Subject: [PATCH] . --- Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Dockerfile b/Dockerfile index bff572148..74dcd2bfc 100644 --- a/Dockerfile +++ b/Dockerfile @@ -77,7 +77,7 @@ WORKDIR /opt/flashinfer ENV FLASHINFER_CUDA_ARCH_LIST="7.5 8.0 8.6 8.9" # 先做 AOT 预编译,再直接打 wheel(不隔离,使用同一份自编 torch) -RUN python3 -m pip install --no-cache-dir numpy requests build "cuda-python>=12.0,<13" "nvidia-nvshmem-cu12" ninja pynvml && \ +RUN python3 -m pip install --no-cache-dir numpy requests build "cuda-python>=12.0,<13" "nvidia-nvshmem-cu12" ninja pynvml filelock && \ python3 -m flashinfer.aot && \ python3 -m build --no-isolation --wheel && \ ls -lh dist/