diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 0ab7cf354bb577b4af365d3b7af174052c67fdfe..efdfd8b588efa108b87fa633582bb8efcf6f4a5a 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -281,7 +281,7 @@ typecheck:
 tests-and-coverage:
   stage: "Unit Tests"
   needs: []
-  image: i10git.cs.fau.de:5005/pycodegen/pycodegen/nox:ubuntu24.04-cuda12
+  image: i10git.cs.fau.de:5005/pycodegen/pycodegen/nox:ubuntu24.04-cuda12.6
   before_script:
     - pip install -e .[tests]
   script:
@@ -311,7 +311,7 @@ tests-and-coverage:
 
 
 build-documentation:
-  image: i10git.cs.fau.de:5005/pycodegen/pycodegen/nox:ubuntu24.04-cuda12
+  image: i10git.cs.fau.de:5005/pycodegen/pycodegen/nox:ubuntu24.04-cuda12.6
   stage: docs
   needs: []
   script: