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.
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?