torch
torchvision
matplotlib
numpy
Pillow
click
