https://singularityfeed.com/use-pytorch-to-easily-access-your-gpu/