torch>=2.6
torchvision
