Commit 75a276b3 authored by Olivier BICHLER's avatar Olivier BICHLER
Browse files

Added CUDA dependencies for aidge_quantization

parent 35ca8d51
Loading
Loading
Loading
Loading
+6 −0
Original line number Diff line number Diff line
@@ -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
          ;;