torch>=1.5.*
torchvision>=0.4.*