torch>=2.3.1
torchvision>=0.18.1