Instructions for installing pytorch to run on GPU - View it on GitHub