diff --git a/Dockerfile b/Dockerfile index 97cfbf9..68c188f 100644 --- a/Dockerfile +++ b/Dockerfile @@ -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,并收集轮子