torch>=1.3.1
torchvision>=0.5.0
numpy>=1.9.1

[examples]
sklearn
matplotlib

[tests]
pytest
