RuntimeError 'CUDA error' when trying the Google Cloud AI Platform Tutorial for PyTorch

Question

I\'ve been trying to get started with Google Cloud\'s AI Platform. I have been following this tutorial: https://cloud.google.com/ai-platform/training/docs/getting-started-pytorc