torch>=2.0.0
torchvision
