From 90e938686c86f2fa95ea035d3a890cea8b72454f Mon Sep 17 00:00:00 2001
From: Wouter Deconinck <wdconinc@gmail.com>
Date: Sun, 26 May 2024 19:22:04 +0000
Subject: [PATCH] fix: use nvidia/cuda image
---
.gitlab-ci.yml | 1 +
1 file changed, 1 insertion(+)
diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 71eecbaf2..8d38349ec 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:
--
GitLab