From 389ea940f333af1879094724f44836c6e2d71d25 Mon Sep 17 00:00:00 2001 From: cmoineau <cyril.moineau@cea.fr> Date: Thu, 24 Apr 2025 12:35:21 +0000 Subject: [PATCH] Use CUDNN image to test backend cuda. --- .gitlab-ci.yml | 23 +++++++++++++++++++++++ 1 file changed, 23 insertions(+) diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml index d82989e..eda42be 100644 --- a/.gitlab-ci.yml +++ b/.gitlab-ci.yml @@ -26,6 +26,29 @@ include: # - '.gitlab/ci/windows_python.gitlab-ci.yml' # - '.gitlab/ci/release/cibuildwheel_windows.gitlab-ci.yml' +build:ubuntu_python: + # Use cudnn image instead of the cuda one + image: nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04 + +test:ubuntu_python: + # Use cudnn image instead of the cuda one + image: nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04 + +coverage:ubuntu_python: + # Use cudnn image instead of the cuda one + image: nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04 + +.build:ubuntu_cpp:template: + # Use cudnn image instead of the cuda one + image: nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04 + +test:ubuntu_cpp: + # Use cudnn image instead of the cuda one + image: nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04 + +coverage:ubuntu_cpp: + # Use cudnn image instead of the cuda one + image: nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04 release:pip:ubuntu: tags: -- GitLab