This commit is contained in:
parent
095427b9d0
commit
70b7f1022b
|
|
@ -77,7 +77,7 @@ WORKDIR /opt/flashinfer
|
||||||
ENV FLASHINFER_CUDA_ARCH_LIST="7.5 8.0 8.6 8.9 9.0a"
|
ENV FLASHINFER_CUDA_ARCH_LIST="7.5 8.0 8.6 8.9 9.0a"
|
||||||
|
|
||||||
# 先做 AOT 预编译,再直接打 wheel(不隔离,使用同一份自编 torch)
|
# 先做 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 flashinfer.aot && \
|
||||||
python3 -m build --no-isolation --wheel && \
|
python3 -m build --no-isolation --wheel && \
|
||||||
ls -lh dist/
|
ls -lh dist/
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue