本教程将指导您如何在您的系统上安装 CUDNN 库。CUDNN 是由 NVIDIA 提供的深度学习加速库,用于提高深度学习应用程序的性能。
安装步骤
下载 CUDNN
访问 NVIDIA CUDNN 下载页面 并根据您的 CUDA 版本选择合适的 CUDNN 版本进行下载。解压文件
将下载的 CUDNN 文件解压到您的系统中的某个目录。配置环境变量
将 CUDNN 的库目录添加到您的LD_LIBRARY_PATH
环境变量中。export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/path/to/cudnn/lib
安装 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
验证安装
运行以下命令验证 CUDNN 是否已正确安装。nvcc --version
如果 CUDNN 已正确安装,您应该看到类似以下输出:
Cuda compiler driver version is cuDNN v8.0.4
注意事项
- 确保您的 CUDA 版本与 CUDNN 版本兼容。
- 如果您在使用 CUDNN 时遇到问题,请查看 NVIDIA CUDNN 文档 获取更多信息。
相关链接
CUDNN