Cuda加速:setUpNet DNN module was not built with CUDA backend; switching to CPU

准备工作(第二步最重要)

  1. 下载opencv 4.4.0以及相应版本的opencv_contrib

  2. 修改cmakelist,在opencv\modules\dnn\CMakeLists.txt文件中添加

    add_definitions(-DHAVE_CUDA=1)

    尝试了很多教程,但是他们都漏了这一个关键的点,因为只有定义了HAVE_CUDA这个宏之后,相应的代码才会被编译进去

编译

参考 linux编译cuda加速版本的opencv
链接里的Step #5很重要

cmake编译选项(仅供参考,自行适配)

cmake \
-D WITH_CUDA=ON \
-D WITH_CUDNN=ON \
-D WITH_V4L=ON \
-D OPENCV_DNN_CUDA=ON \
-D CUDNN_VERSION=‘8.0’ \
-D CUDNN_INCLUDE_DIR=’/usr/include/’ \
-D ENABLE_FAST_MATH=1 \
-D CUDA_FAST_MATH=1 \
-D CUDA_ARCH_BIN=“5.3” \
-D CUDA_ARCH_PTX="" \
-D OPENCV_EXTRA_MODULES_PATH=./opencv_contrib/modules \
-D WITH_GSTREAMER=ON \
-D WITH_LIBV4L=ON \
-D BUILD_opencv_python2=OFF \
-D BUILD_opencv_python3=ON \
-D BUILD_TESTS=OFF \
-D BUILD_PERF_TESTS=OFF \
-D BUILD_EXAMPLES=OFF \
-D CMAKE_BUILD_TYPE=RELEASE \
-D CMAKE_INSTALL_PREFIX=/usr/local \
..

注意换行符后面不能有空格及任何字符,结尾的..不能省略

评论 15
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值