pytorch-lightning>=1.5.0
torch>=1.8.0
