torch==2.0.1 torchvision==0.15.2 numpy==1.22.4 scikit-learn==1.2.2 plotly==5.13.1