torch==2.5.1
numpy==2.2.1
torchvision==0.20.1