packaging
pip
torch>=1.5.0
torchinfo>=0.0.9
torchvision>=0.6

[dev]
tensorboard
