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