.

2025-07-02 20:08:05 +08:00 · 2025-07-02 20:08:05 +08:00 · 49b5e17996
parent 32f02ee68e
commit 49b5e17996
1 changed files with 1 additions and 2 deletions
--- a/3
+++ b/3
@ -31,8 +31,7 @@ RUN git clone --recursive -b v2.7.1 https://github.com/pytorch/pytorch.git
 WORKDIR /opt/pytorch
 ENV MAX_JOBS=${MAX_JOBS}
 RUN echo "Building PyTorch with USE_DISTRIBUTED=$USE_DISTRIBUTED" && \
-    python3 setup.py bdist_wheel && \
-    python3 -c "from torch.distributed import Backend; print('✅ Build success. GLOO =', Backend.GLOO)"
+    python3 setup.py bdist_wheel

 ###############################################################################
 # Stage 1 ─ builder-extras：用自编 Torch 装 TV / flashinfer / sglang，并收集轮子