This commit is contained in:
hailin 2025-07-25 12:32:56 +08:00
parent 35ba2eab42
commit 6ea2139b82
1 changed files with 2 additions and 1 deletions

View File

@ -110,7 +110,7 @@ RUN pip wheel \
# ── ✅ 打包 gradio UI 所需依赖 ──────────────────────────────────────────────── # ── ✅ 打包 gradio UI 所需依赖 ────────────────────────────────────────────────
RUN pip wheel gradio requests -w /wheels RUN pip wheel gradio requests -w /wheels
############################################################################### ###############################################################################
# Stage 2 ─ runtime极简运行镜像仅离线安装 wheel # Stage 2 ─ runtime极简运行镜像仅离线安装 wheel
############################################################################### ###############################################################################
@ -140,6 +140,7 @@ COPY --from=builder-extras /tmp/sgl_kernel_wheel /tmp/sgl_kernel_wheel
# ✅ 优先装你自编的 torch避免被 PyPI 上的覆盖 # ✅ 优先装你自编的 torch避免被 PyPI 上的覆盖
RUN ls -lh /tmp/wheels && \ RUN ls -lh /tmp/wheels && \
rm -f /tmp/wheels/torch-2.7.1a0+*.whl && \ rm -f /tmp/wheels/torch-2.7.1a0+*.whl && \
rm -f /tmp/wheels/huggingface_hub-0.33.4*.whl && \
python3 -m pip install --no-cache-dir --no-deps /tmp/wheels/torch*.whl && \ python3 -m pip install --no-cache-dir --no-deps /tmp/wheels/torch*.whl && \
python3 -m pip install --no-cache-dir --no-deps /tmp/wheels/vllm-*.whl && \ python3 -m pip install --no-cache-dir --no-deps /tmp/wheels/vllm-*.whl && \
python3 -m pip install --no-cache-dir --no-deps /tmp/sgl_kernel_wheel/*.whl && \ python3 -m pip install --no-cache-dir --no-deps /tmp/sgl_kernel_wheel/*.whl && \