# Constraints to prevent pip from upgrading PyTorch during requirements install
# The base image rocm/pytorch:rocm6.4_ubuntu22.04_py3.10_pytorch_release_2.5.1 has:
# - torch 2.5.1+git8420923 (ROCm 6.4 build)
# - torchvision 0.20.1a0+04d8fc4

# Pin torch to exact version from base image to prevent upgrades
torch==2.5.1+git8420923
torchvision==0.20.1a0+04d8fc4

# Prevent standard torchaudio (we'll install ROCm version separately)
torchaudio