From 75a276b3685da719efe1bee4909a30bf92fb4cf0 Mon Sep 17 00:00:00 2001
From: Olivier BICHLER <olivier.bichler@cea.fr>
Date: Tue, 26 Nov 2024 09:03:16 +0000
Subject: [PATCH] Added CUDA dependencies for aidge_quantization
---
.gitlab/ci/default.gitlab-ci.yml | 6 ++++++
1 file changed, 6 insertions(+)
diff --git a/.gitlab/ci/default.gitlab-ci.yml b/.gitlab/ci/default.gitlab-ci.yml
index 19d311d..df934d8 100644
--- a/.gitlab/ci/default.gitlab-ci.yml
+++ b/.gitlab/ci/default.gitlab-ci.yml
@@ -51,6 +51,12 @@ variables:
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDACXX=$(which nvcc)
;;
+ "aidge_quantization" )
+ apt-get install -y -qq libcudnn8-dev
+ export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
+ export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
+ export CUDACXX=$(which nvcc)
+ ;;
"aidge_backend_opencv" )
apt-get install -y -qq libopencv-dev
;;
--
GitLab