torch
matplotlib
torchvision