pytorch-lightning>=1.7.0, <=1.7.6
torch>=1.9.*
