build.sh 997 B

123456789101112131415161718192021222324252627
  1. #!/usr/bin/env bash
  2. # Build GLM-5.2 NF3-Hybrid Docker image for DGX Spark (aarch64/sm_121a)
  3. # All build inputs are version-pinned in versions.env and locks/.
  4. set -euo pipefail
  5. cd "$(dirname "$0")"
  6. TAG="${TAG:-vllm-nf3-hybrid:probe}"
  7. source versions.env
  8. ARGS=""
  9. for v in CUDA_BASE_IMAGE CUDA_BASE_DIGEST UV_VERSION \
  10. TORCH_VERSION TORCHVISION_VERSION TORCHAUDIO_VERSION TRITON_VERSION \
  11. PYTORCH_INDEX_URL PYPI_INDEX_URL \
  12. NVSHMEM_VERSION TVM_FFI_VERSION TILELANG_VERSION NUMBA_VERSION \
  13. NCCL_REPO NCCL_REF \
  14. VLLM_REPO VLLM_REF VLLM_COMMIT \
  15. B12X_REPO B12X_REF B12X_COMMIT \
  16. FLASHINFER_REPO FLASHINFER_REF \
  17. CUTLASS_DSL_VERSION \
  18. TORCH_CUDA_ARCH_LIST FLASHINFER_CUDA_ARCH_LIST GB10_BUILD; do
  19. val="${!v:-}"
  20. [ -n "$val" ] && ARGS="$ARGS --build-arg ${v}=${val}"
  21. done
  22. echo "=== Building ${TAG} ==="
  23. DOCKER_BUILDKIT=1 docker build $ARGS --progress=plain -t "${TAG}" -f Dockerfile . 2>&1
  24. echo "=== Done: ${TAG} ==="