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