-i https://download.pytorch.org/whl/rocm7.1
torch==2.10.*
torchvision==0.25.*