torch>=1.9.0
numpy>=1.19.0
torchvision>=0.10.0
pandas>=1.3.0
matplotlib>=3.3.0

[dev]
pytest>=6.0
pytest-cov>=2.12
black>=21.0
flake8>=3.9
mypy>=0.910

[gpu]
pynvml>=11.0.0

[notebooks]
jupyter>=1.0.0
