c++ - 使用Cuda Ubuntu 16/04构建C++ OpenCV 4

Question

我正在尝试使用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 10.2相对应的cudnn7，我的安装测试通过了。

有人可以帮忙吗？

Answer 1

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

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