Commit 56873fbc authored by Olivier BICHLER's avatar Olivier BICHLER
Browse files

Added aidge_pruning

parent d6844fbd
Loading
Loading
Loading
Loading
+6 −0
Original line number Diff line number Diff line
@@ -57,6 +57,12 @@ variables:
          export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
          export CUDACXX=$(which nvcc)
          ;;
        "aidge_pruning" )
          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
          ;;