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