pytorch-lightning>=1.6.1
torchvision>=0.12.0
torch>=1.11.0
protobuf>=3.19.0,<=3.20.0
