torchvision
matplotlib
torch
