"Ubuntu NVIDIA GPU-Optimized Image" doesn't support PyTorch 2.11

[oscarkey]

Type of issue
Request for additional tools or software for an existing image.

Describe the issue
PyTorch 2.11 switches to cuda 13.0 by default. This requires cuda driver >= 580, which I guess is newer than on this image, and we get The NVIDIA driver on your system is too old.

You can probably work around this by installing pytorch with an older version of cuda, but it's not ideal.

To Reproduce

1. Create a runner in this configuration

2. Try to run a workflow that installs pytorch 2.11 from pypi and runs it
3. Get error The NVIDIA driver on your system is too old

Desktop (please complete the following information):

• OS: NVIDIA GPU-Optimized Image for AI and HPC