# PREREG-STAGE2 AMENDMENT 3 box (A3.1): torch 2.6.0 cu124 wheels (released 2025-01) + matching triton, CUDA 12.4 nvcc/headers.
# Installed into a fresh venv /opt/kbvpa/v (system python3 = 3.12 on gpu-base-24-04, asserted), WITHOUT system site-packages.
# torch's own metadata pins its nvidia-*-cu12 runtime wheels exactly (cuda-runtime/nvrtc/cupti 12.4.127, cublas, cudnn, ...);
# pip freeze is saved to setup/packages.txt on the box.
--extra-index-url https://download.pytorch.org/whl/cu124
torch==2.6.0+cu124
triton==3.2.0
# torch 2.6.0 is built against the NumPy 2 ABI and supports numpy 2.x; no numpy<2 pin is needed. 2.2.x = the 2025 series.
numpy>=2.2,<2.3
# Headers for torch.utils.cpp_extension.load_inline, same CUDA 12.4 release as the runtime (setup_box.sh asserts nvcc
# major.minor == torch.version.cuda). cccl: cub/thrust/nv/. Fix round 1 (BUILD REVIEW A3 B1): the nvcc wheel 12.4.x holds only
# ptxas, include/crt and libnvvm, NO bin/nvcc; it stays installed but is not used. nvcc comes from NVIDIA's CUDA 12.4.1 redist
# archive (sha256-pinned in setup_box.sh) via make_cuda_home.py --nvcc-root.
nvidia-cuda-nvcc-cu12>=12.4,<12.5
nvidia-cuda-cccl-cu12>=12.4,<12.5
ninja==1.13.2
packaging==26.3
