torchvision >= 0.17.0
