From 2e45dc43461a195b61ab832419dca7f92e80157e Mon Sep 17 00:00:00 2001
From: Stefan Reimer <stefan@zero-downtime.net>
Date: Fri, 8 Nov 2024 19:36:53 +0000
Subject: [PATCH] ci: update nvidia test pod

---
 docs/nvidia-test.yaml | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/docs/nvidia-test.yaml b/docs/nvidia-test.yaml
index ac8f8e74..87000152 100644
--- a/docs/nvidia-test.yaml
+++ b/docs/nvidia-test.yaml
@@ -4,8 +4,9 @@ metadata:
   name: nvidia-test
   namespace: default
 spec:
+  runtimeClassName: nvidia
   containers:
-  - image: nvidia/cuda:10.1-runtime-ubuntu16.04
+  - image: nvidia/cuda:12.5.1-base-ubuntu22.04
     command:
     - "bash"
     - "-c"