diff --git a/Dockerfile b/Dockerfile index 0461e8a63..906e56663 100644 --- a/Dockerfile +++ b/Dockerfile @@ -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 ###############################################################################