torch
numpy
torchvision
matplotlib
