torch
torchvision
numpy
tqdm
omegaconf>=2.0.0
pytorch-lightning>=1.0.8
