This commit is contained in:
hailin 2025-07-02 14:00:20 +08:00
parent ce0ca623c8
commit d53a433b62
1 changed files with 4 additions and 4 deletions

View File

@ -56,8 +56,8 @@ RUN python3 setup.py bdist_wheel
# ── 编译 flashinfer (主分支支持 torch 2.7 / cu126) ─────────────────────────
WORKDIR /opt
RUN git clone https://github.com/Dao-AILab/flashinfer.git
WORKDIR /opt/flashinfer/python
RUN git clone https://github.com/flashinfer-ai/flashinfer.git
WORKDIR /opt/flashinfer
RUN python3 setup.py bdist_wheel
# ── 编译你本地 sglang 源码并打 wheel ───────────────────────────────────────
@ -69,8 +69,8 @@ RUN python3 -m pip install ".[srt,openai]" --no-build-isolation && \
# ── 收集所有 wheel 到 /wheels ──────────────────────────────────────────────
RUN mkdir -p /wheels && \
cp /tmp/torch_dist/torch*.whl /wheels/ && \
cp /opt/vision/dist/torchvision-0.22.1+cu126*.whl /wheels/ && \
cp /opt/flashinfer/python/dist/flashinfer_python-*.whl /wheels/ && \
cp /opt/vision/dist/torchvision-*.whl /wheels/ && \
cp /opt/flashinfer/dist/flashinfer_python-*.whl /wheels/ && \
cp /tmp/sg_wheels/sglang-*.whl /wheels/
###############################################################################