imageio
matplotlib
numpy>=1.17
torch
tqdm
