tensorflow
torchvision
numpy
matplotlib
torch