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

[dev]
tensorboard
