PyTorch 1.7.0+
torchvision 0.8.1+
numpy