packages needed to run this program - pytorch, torchvision, numpy, matlotlib, ipython which can be installed by running the following command from your terminal
pip install torch torchvision matplotlib ipython numpy
after installing the packages ou can see the ai doing its magic by running the agent.py program
you should let the program run for 10 mins to start getting accurate results
you can also the play the game yourself by running the snake.py program