diff --git a/.gitlab/ci/default.gitlab-ci.yml b/.gitlab/ci/default.gitlab-ci.yml index 19d311da69003983aab3f0b178d8e1a09e68a243..df934d846fecbe6beffa2fba50097f7d24a7b494 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 ;;