Small script to test that PyTorch works with the current GPU(s). - View it on GitHub