torch
torchvision
psutil
matplotlib
numpy