ci: update nvidia test pod

2024-11-08 19:36:53 +00:00 · 2024-11-08 19:36:53 +00:00 · 2e45dc4346
commit 2e45dc4346
parent 9504f874a0
1 changed files with 2 additions and 1 deletions
--- 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"