torch>=1.4.0
torchvision>=0.5.0
numpy

[dev]
Sphinx
sphinx_rtd_theme
