This commit is contained in:
parent
6713da1fb6
commit
51f99ac18f
|
|
@ -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
|
||||
###############################################################################
|
||||
|
|
|
|||
Loading…
Reference in New Issue