torch==2.0.1
torchvision==0.15.2
matplotlib==3.7.1
scikit-learn==1.2.2
seaborn==0.12.2
numpy==1.24.3
pillow