什么是 cuDNN?
cuDNN(CUDA Deep Neural Network)是 NVIDIA 提供的深度学习加速库,与 OpenCV 深度学习模块结合可显著提升 GPU 计算效率。
配置步骤
环境准备
- 确保已安装 CUDA Toolkit
- 下载与 CUDA 版本匹配的 cuDNN 安装包
- 验证 GPU 驱动版本(建议 ≥ 418.34)
安装 cuDNN
# 解压安装包示例 tar -xvf cudnn${version}_linux_${cuda_version}_${install_type}_v${cuDNN_version}.tar.gz
📌 将解压后的文件复制到 CUDA 安装目录(如
/usr/local/cuda
)配置环境变量
修改~/.bashrc
或系统环境变量文件:export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH export PATH=/usr/local/cuda/bin:$PATH
注意事项⚠️
- 版本兼容性:CUDA + cuDNN 版本对照表
- 避免与 OpenCV 的
dnn
模块冲突(建议使用 OpenCV 4.5.0+) - 安装后运行
nvcc --version
验证是否生效
扩展学习
想深入了解 OpenCV 深度学习框架?
👉 点击此处查看 OpenCV 深度学习教程总览