使用 CUDA Ubuntu 16/04 构建 C++ OpenCV 4

building c++ opencv 4 with cuda ubuntu 16/04

本文关键字:构建 C++ OpenCV CUDA Ubuntu 使用      更新时间:2023-10-16

我正在尝试使用 cuda 10.2 构建 opencv'。当以下命令:

cmake -DCMAKE_BUILD_TYPE=RELEASE 
-DOPENCV_EXTRA_MODULES_PATH=../../opencv_contrib/modules 
-DWITH_TBB=ON -DWITH_CUDA=ON 
-DBUILD_opencv_cudacodec=OFF 
-DENABLE_FAST_MATH=1 
-DWITH_CUBLAS=1 
-DWITH_V4L=ON  
-DWITH_OPENGL=ON  
-DWITH_GSTREAMER=ON  
-DOPENCV_GENERATE_PKGCONFIG=ON  
-DOPENCV_ENABLE_NONFREE=ON 
-DBUILD_EXAMPLES=TRUE 
-DBUILD_PERF_TESTS=FALSE 
-DEBUILD_TESTS=FALSE ../../opencv

我有以下问题:

Could NOT find CUDNN (missing:  CUDNN_LIBRARY CUDNN_INCLUDE_DIR) (Required is at least version "6")

当然,我已经安装了与cuda 7对应的cudnn10.2,我的安装测试通过了。

有人可以帮忙吗?

为了解决这个问题,我为 cmake 添加了以下选项:

-DCUDNN_INCLUDE_DIR=/usr/local/cuda/include 
-DCUDNN_LIBRARY=/usr/local/cuda/lib64/libcudnn.so.7.6.5 

您可能还需要添加:-DCUDNN_VERSION='7.6'

或者就我而言,它是:-DCUDNN_VERSION='8.0'