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: