ipykernel
jupyter
matplotlib
numpy
torch
torchvision
scikit-learn
pandas
PyYAML