torch>=1.7.0
numpy

[test]
pytest>=6.0
torchvision
tqdm
