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