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