This commit is contained in:
hailin 2025-09-22 12:15:38 +08:00
parent 6713da1fb6
commit 51f99ac18f
1 changed files with 4 additions and 0 deletions

View File

@ -41,6 +41,8 @@ RUN echo "Building PyTorch with USE_DISTRIBUTED=$USE_DISTRIBUTED" && \
export PYTORCH_BUILD_VERSION=2.8.0 PYTORCH_BUILD_NUMBER=1 && \
python3 setup.py bdist_wheel
###############################################################################
# Stage 1 ─ builder-extras用自编 Torch 装 TV / flashinfer / sglang并收集轮子
###############################################################################
@ -209,6 +211,8 @@ RUN pip wheel pybase64==1.3.2 -w /wheels
FROM scratch AS wheelhouse
COPY --from=builder-extras /wheels /
###############################################################################
# Stage 2 ─ runtime极简运行镜像仅离线安装 wheel
###############################################################################