pygame==2.1.2 torch torchvision scikit-learn==1.0.2 pandas==1.4.2