Skip to content
Snippets Groups Projects
Commit 75a276b3 authored by Olivier BICHLER's avatar Olivier BICHLER
Browse files

Added CUDA dependencies for aidge_quantization

parent 35ca8d51
No related branches found
No related tags found
No related merge requests found
......@@ -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
;;
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment