torch
torchvision
pandas
matplotlib
numpy
tqdm
notebook
scikit-learn