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