pygame==2.1.2
torch
torchvision