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