Ubuntu下OpenCV编译找不到cuDNN库文件怎么办？

在Ubuntu下编译OpenCV时，如果遇到找不到cuDNN库文件的问题，通常是因为CMake在配置过程中无法正确找到cuDNN的路径。以下是一些可能的解决方案：

1. 确保cuDNN已正确安装

首先，确保cuDNN已经正确安装。你可以通过以下命令检查cuDNN的安装路径：

find /usr -name "libcudnn*"

如果cuDNN已经安装，你应该能够看到类似/usr/local/cuda/lib64/libcudnn.so的路径。

2. 设置环境变量

确保CUDA_HOME和CUDNN_ROOT环境变量已正确设置。你可以在~/.bashrc或~/.zshrc中添加以下内容：

export CUDA_HOME=/usr/local/cuda
export CUDNN_ROOT=/usr/local/cuda
export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH

然后运行source ~/.bashrc或source ~/.zshrc使更改生效。

3. 手动指定cuDNN路径

在CMake配置时，手动指定cuDNN的路径。你可以使用以下命令：

cmake -D CMAKE_BUILD_TYPE=RELEASE \
      -D CMAKE_INSTALL_PREFIX=/usr/local \
      -D OPENCV_EXTRA_MODULES_PATH=../../opencv_contrib/modules \
      -D WITH_CUDA=ON \
      -D WITH_CUDNN=ON \
      -D OPENCV_DNN_CUDA=ON \
      -D CUDA_ARCH_BIN=7.5 \
      -D CUDA_ARCH_PTX=7.5 \
      -D CUDA_FAST_MATH=ON \
      -D WITH_CUBLAS=ON \
      -D CUDNN_INCLUDE_DIR=/usr/local/cuda/include \
      -D CUDNN_LIBRARY=/usr/local/cuda/lib64/libcudnn.so \
      ..

4. 检查CMake输出

在运行CMake时，仔细检查输出，确保cuDNN被正确找到。你应该看到类似以下的输出：

-- Found CUDNN: /usr/local/cuda/lib64/libcudnn.so

如果没有找到，可能需要重新检查路径设置。

5. 更新CMake版本

如果你使用的是较旧的CMake版本，可能会遇到一些问题。尝试更新到最新版本的CMake：

sudo apt-get update
sudo apt-get install --only-upgrade cmake

6. 检查CUDA和cuDNN版本兼容性

确保你安装的CUDA和cuDNN版本是兼容的。你可以参考NVIDIA官方文档来确认版本兼容性。

7. 重新编译OpenCV

如果以上步骤都正确配置，尝试重新编译OpenCV：

make -j$(nproc)
sudo make install

8. 检查系统路径

确保/usr/local/cuda/lib64在系统的库路径中。你可以通过以下命令检查：

echo $LD_LIBRARY_PATH

如果没有包含/usr/local/cuda/lib64，可以手动添加：

export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

总结

通过以上步骤，你应该能够解决在Ubuntu下编译OpenCV时找不到cuDNN库文件的问题。如果问题仍然存在，建议检查CUDA和cuDNN的安装是否正确，并确保所有路径设置无误。