torch
torchvision
matplotlib

[dev]
pytest>=6.0
