Update nvidia/cuda Docker tag to v12.6.1

gruberdev · Sep 5, 2024 · 0e16d6e · 0e16d6e
1 parent 9be261a
commit 0e16d6e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tasks/test.yaml b/tasks/test.yaml
@@ -53,7 +53,7 @@ tasks:
           restartPolicy: Never
           containers:
             - name: gpu
-              image: "nvidia/cuda:12.6.0-base-ubuntu20.04@sha256:b0937b96f04e1f73279ead20a1d3bb38fb569071756ce1cbde337616beb63129"
+              image: "nvidia/cuda:12.6.1-base-ubuntu20.04@sha256:5d012529898dc2dc6f79671e7f77ab48459d7c31a73dff259b274afa80069e9d"
               command: [ "/bin/bash", "-c", "--" ]
               args: [ "while true; do sleep 30; done;" ]
               resources: