Use correct CUDA container references as noted in the dockerhub overview

This commit is contained in:
Simon Redman 2025-04-27 21:16:03 -04:00
parent 0affde29e6
commit e65958ebe2
No known key found for this signature in database
GPG key ID: 2B14B66FAEFCDEFE

View file

@ -64,7 +64,7 @@ To check what CUDA version do you need, you can either run `nvidia-smi` or `nvcc
Alternatively, you can also check nvidia-smi with docker:
```
docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi
docker run --runtime=nvidia --rm nvidia/cuda:12.8.0-base-ubuntu24.04 nvidia-smi
```
To use CUDA, use the images with the `cublas` tag, for example.