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