Sorry if this is a newb question.  I don’t have a lot of experience with GPUs…
I created a GCE instance like this:
gcloud compute instances create my-instance --image-family=pytorch-latest-gpu ...

The instance that spins up has torch 1.4 instead of the latest one.  With this version, torch.cuda.is_available() returns True.
After doing pip install -U torch to get the latest version, torch.cuda.is_available() returns False.
What is the best way to get a working torch==1.6.0 instance on GCE?