versions.env 1.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243
  1. # versions.env - authoritative build inputs for Solar Open2 250B NVFP4 (Nota)
  2. # Every entry is a tag or exact version - no :latest, no main, no ranges.
  3. #
  4. # Stack mirrors glm52-nf3-hybrid (cu130 / aarch64 / sm_121a) minus the b12x
  5. # suite: Solar is a stock vLLM serve using the UpstageAI solar_open2 fork
  6. # (solar_open2 reasoning/tool parsers + SolarOpen2TemplateLogitsProcessor).
  7. # NVFP4 (compressed-tensors) is auto-detected from the model config.
  8. #
  9. # locks/ and checksums/ are shared with glm52-nf3-hybrid (same base stack).
  10. CUDA_BASE_IMAGE=nvidia/cuda:13.2.0-devel-ubuntu24.04
  11. CUDA_BASE_DIGEST=sha256:a5b6256e470196fc1d5f8f62139d57d3662867746dfe1cb352d7652024047020
  12. # Python / uv
  13. UV_VERSION=0.11.28
  14. # PyTorch stack (cu130 index works on CUDA 13.2 for aarch64)
  15. TORCH_VERSION=2.11.0
  16. TORCHVISION_VERSION=0.26.0
  17. TORCHAUDIO_VERSION=2.11.0
  18. TRITON_VERSION=3.6.0
  19. PYTORCH_INDEX_URL=https://download.pytorch.org/whl/cu130
  20. PYPI_INDEX_URL=https://pypi.org/simple
  21. # NCCL
  22. NCCL_REPO=https://github.com/NVIDIA/nccl.git
  23. NCCL_REF=v2.30.7-1
  24. # vLLM (UpstageAI fork — solar_open2 parsers + logits processor)
  25. VLLM_REPO=https://github.com/UpstageAI/vllm.git
  26. VLLM_REF=v0.22.0-solar-open2
  27. VLLM_COMMIT=00907fc9b982f973808a0193ad0f8c61fbb62854
  28. # FlashInfer
  29. FLASHINFER_REPO=https://github.com/flashinfer-ai/flashinfer.git
  30. FLASHINFER_REF=v0.6.12
  31. # cutlass-dsl (4.5.3 for firmware 580.159.03 compat)
  32. CUTLASS_DSL_VERSION=4.5.3
  33. # Target GPU architecture (DGX Spark = sm_121a)
  34. TORCH_CUDA_ARCH_LIST=12.1a
  35. FLASHINFER_CUDA_ARCH_LIST=12.1a