pillow
scipy
pytorch
cudatoolkit
tqdm
imageio