pytorch-lightning<=1.9.5
torch
datasets
