torch>=1.6
gym
pyscreenshot
