torchvision==0.11.1
torch==1.10.0
matplotlib==3.4.3
wandb==0.12.7
pytorch-fid==0.2.1
pynvml==11.0.0