I’m using a V100 GPU.
I’m guessing that torch 1.4 and 1.6 use different CUDA versions so that the previously installed drivers don’t work with torch 1.6.
It is a bummer that Google isn’t updating the pytorch-latest-gpu image.