torch
torchvision
torchaudio
numpy
matplotlib