CUDNN（CUDA Deep Neural Network library）是 NVIDIA 推出的一套深度学习加速库，它提供了深度神经网络在 NVIDIA GPU 上的高效实现。

安装步骤

  1. 下载 CUDNN
    访问 NVIDIA 官方网站 下载适合您 CUDA 版本的 CUDNN。

  2. 解压文件
    将下载的 CUDNN 文件解压到一个合适的位置。

  3. 配置环境变量
    将 CUDNN 的 include 和 lib 目录添加到您的 LD_LIBRARY_PATH 和 C_INCLUDE_PATH 环境变量中。

    export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/path/to/cudnn/lib
export C_INCLUDE_PATH=$C_INCLUDE_PATH:/path/to/cudnn/include

  4. 安装 CUDNN
    根据您的操作系统和编译器，运行相应的安装脚本。

    sudo ./install.sh

  5. 验证安装
    使用以下命令验证 CUDNN 是否正确安装：

    nvcc --version

    如果输出中包含 CUDNN 的版本信息，则表示安装成功。

扩展阅读

如果您需要更详细的安装步骤或遇到安装问题，可以参考 本站关于 CUDNN 安装的详细教程。

相关资源

CUDNN