diff --git a/tasks/test.yaml b/tasks/test.yaml
index 51da9ae09..008ceba92 100644
--- 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: