pytorch-lightning>=1.8.0, <=1.8.2
torch>=1.9.*
