This commit is contained in:
parent
ce0ca623c8
commit
d53a433b62
|
|
@ -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/
|
||||
|
||||
###############################################################################
|
||||
|
|
|
|||
Loading…
Reference in New Issue