diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 9f80cd261f75d6677ddf70f28331cd99f4fa7397..13f4592f3df4ad98dc80c5256a326e1e950fe602 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -286,7 +286,7 @@ mypy-typecheck:
 tests-and-coverage:
   stage: "Unit Tests"
   needs: []
-  image: i10git.cs.fau.de:5005/pycodegen/pycodegen/full
+  image: i10git.cs.fau.de:5005/pycodegen/pycodegen/full:cuda11
   before_script:
     - pip install -e .[tests]
   script:
@@ -318,7 +318,7 @@ tests-and-coverage:
 
 
 build-documentation:
-  image: i10git.cs.fau.de:5005/pycodegen/pycodegen/full
+  image: i10git.cs.fau.de:5005/pycodegen/pycodegen/full:cuda11
   stage: docs
   needs: []
   before_script: