torch >= 2.0
torchvision >= 0.15
