torch==2.3.1
torchvision==0.18.1