.

2025-09-17 08:23:15 +08:00 · 2025-09-17 08:23:15 +08:00 · 70b7f1022b
parent 095427b9d0
commit 70b7f1022b
1 changed files with 1 additions and 1 deletions
--- a/2
+++ b/2
@ -77,7 +77,7 @@ WORKDIR /opt/flashinfer
 ENV FLASHINFER_CUDA_ARCH_LIST="7.5 8.0 8.6 8.9 9.0a"

 # 先做 AOT 预编译，再直接打 wheel（不隔离，使用同一份自编 torch）
-RUN python3 -m pip install --no-cache-dir numpy requests build && \
+RUN python3 -m pip install --no-cache-dir numpy requests build "cuda-python>=12.0,<13" && \
    python3 -m flashinfer.aot && \
    python3 -m build --no-isolation --wheel && \
    ls -lh dist/