From e65958ebe2b5767e86caae48d941b3f7126ca01d Mon Sep 17 00:00:00 2001 From: Simon Redman Date: Sun, 27 Apr 2025 21:16:03 -0400 Subject: [PATCH] Use correct CUDA container references as noted in the dockerhub overview --- docs/content/docs/features/GPU-acceleration.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/content/docs/features/GPU-acceleration.md b/docs/content/docs/features/GPU-acceleration.md index 811cfff3..a92a8b3c 100644 --- a/docs/content/docs/features/GPU-acceleration.md +++ b/docs/content/docs/features/GPU-acceleration.md @@ -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.