torch
torchvision
ipython
pillow
matplotlib
numpy