GPUs not available inside containers

[vedantroy-genmo]

I'm using GCP batch with GPUs, but I'm getting errors like the following:

```

assert torch.cuda.is_available(), "GPU not available"

```

or:

```

RuntimeError: Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and installed a driver from http://www.nvidia.com/Download/index.aspx

```

I am running my jobs inside of containers. And, I'm installing the versions of Pytorch, etc. that have CUDA drivers.

Here is my job config: https://paste.gg/p/anonymous/1678edc73dde45459ade23caa76ec260. The job ID is: process-data-id28-3785087a-13a1-46ba00.