使用 cuda ubuntu 构建 c++ opencv 4 16/04

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 \

Answer 2

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

或者在我的情况下是： -DCUDNN_VERSION='8.0'