CUDA images and drivers

Every GPU node ships with CUDA 12.2, cuDNN 8.9 and PyTorch 2.3 preinstalled.

Verify the driver

$ nvidia-smi
$ python -c "import torch; print(torch.cuda.is_available())"

If the second command prints False, reboot once and contact support.