torch>=1.7.0
tensorboard
pytorch-fid
PyYAML
mpi4py
scipy
numpy
lmdb
tqdm
