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

在Ubuntu下编译OpenCV时，如果遇到找不到cuDNN库文件的问题，可以按照以下步骤进行排查和解决：

1. 确认cuDNN已正确安装

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

find /usr/local -name "libcudnn*"

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

2. 设置环境变量

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

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

然后执行以下命令使更改生效：

source ~/.bashrc  # 或者 source ~/.zshrc

3. 检查CMake配置

在编译OpenCV时，确保CMake配置正确指向cuDNN库。你可以通过以下命令检查CMake配置：

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 \  # 根据你的GPU架构调整
      -D CUDA_FAST_MATH=ON \
      -D WITH_CUBLAS=ON \
      -D CUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda \
      -D CUDA_CUDA_LIBRARY=/usr/local/cuda/lib64/stubs/libcuda.so \
      -D CUDA_NVCC_FLAGS="--expt-relaxed-constexpr" \
      -D CUDNN_INCLUDE_DIR=/usr/local/cuda/include \
      -D CUDNN_LIBRARY=/usr/local/cuda/lib64/libcudnn.so \
      ..

确保CUDNN_INCLUDE_DIR和CUDNN_LIBRARY指向正确的cuDNN路径。

4. 检查cuDNN版本兼容性

确保你安装的cuDNN版本与CUDA版本兼容。你可以通过以下命令检查CUDA和cuDNN的版本：

nvcc --version  # 检查CUDA版本
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2  # 检查cuDNN版本

5. 重新编译OpenCV

在确认以上配置无误后，重新运行CMake并编译OpenCV：

cmake ..
make -j$(nproc)
sudo make install

6. 检查编译日志

如果问题仍然存在，检查编译日志以获取更多信息。通常，编译日志会明确指出缺少的库文件或路径问题。

7. 安装缺失的依赖

如果编译日志提示缺少某些依赖库，可以使用以下命令安装：

sudo apt-get install libcudnn8 libcudnn8-dev

8. 使用预编译的OpenCV

如果你不想手动编译OpenCV，可以考虑使用预编译的OpenCV包：

sudo apt-get install libopencv-dev

总结

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