Solved by ptrblck in post #3 
              
              
                Yes, you should be able to use a CUDA10.2 docker container, as its driver requirement would be met by the newer CUDA11 driver.
I don’t know what a Google machine image is and how hard it would be to build it.
You could reuse the Dockerfile from the PyTorch repository.

Also, did you try to install …