matplotlib
imageio
numpy
torch==1.13.1
tqdm
