matplotlib
numpy
pandas
torch>=1.10
torchvision
scikit-learn
csv
