torch==2.8.0
torchvision==0.23.0