matplotlib
torch
Pillow