本教程将指导您如何在您的系统上安装 CUDNN 库。CUDNN 是由 NVIDIA 提供的深度学习加速库,用于提高深度学习应用程序的性能。

安装步骤

  1. 下载 CUDNN
    访问 NVIDIA CUDNN 下载页面 并根据您的 CUDA 版本选择合适的 CUDNN 版本进行下载。

  2. 解压文件
    将下载的 CUDNN 文件解压到您的系统中的某个目录。

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

    export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/path/to/cudnn/lib
    
  4. 安装 CUDNN
    运行以下命令安装 CUDNN。

    sudo cp /path/to/cudnn/lib/libcudnn* /usr/local/cuda/lib64
    sudo cp /path/to/cudnn/include/cudnn.h /usr/local/cuda/include
    
  5. 验证安装
    运行以下命令验证 CUDNN 是否已正确安装。

    nvcc --version
    

    如果 CUDNN 已正确安装,您应该看到类似以下输出:

    Cuda compiler driver version is cuDNN v8.0.4
    

注意事项

  • 确保您的 CUDA 版本与 CUDNN 版本兼容。
  • 如果您在使用 CUDNN 时遇到问题,请查看 NVIDIA CUDNN 文档 获取更多信息。

相关链接

CUDNN