pygame==2.1.2 torch torchvision scikit-learn==1.0.2 pandas==1.4.2 numpy~=1.22.3 matplotlib~=3.5.1