pillow
scipy
pytorch
cudatoolkit
tqdm=4.62.2
imageio