diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 71eecbaf261b40cacf34b8967e7368e161e1d421..8d38349ec78b1429d2f6644a7dcf9ceaf3641cc6 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -90,6 +90,7 @@ workflow:
 
 nvidia-smi:
   stage: config
+  image: nvidia/cuda:11.3.1-base-ubuntu20.04
   tags:
     - gpu
   script: