cudnn在ubuntu上如何配置

要在Ubuntu上配置cuDNN，首先需要确保已经安装了NVIDIA的GPU驱动和CUDA。然后按照以下步骤进行配置：

1. 下载cuDNN：首先需要登录NVIDIA的开发者网站（https://developer.nvidia.com/cudnn），注册并下载适用于你的CUDA版本的cuDNN压缩包。

2. 解压缩cuDNN：解压下载的压缩包，并将其中的文件复制到CUDA的安装目录下。一般情况下，CUDA的安装路径为/usr/local/cuda/。

3. 设置环境变量：编辑~/.bashrc文件，并添加以下几行代码：

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"
export CUDA_HOME=/usr/local/cuda

保存并退出文件，然后运行以下命令使配置生效：

source ~/.bashrc

4. 检查cuDNN是否配置成功：在终端中输入以下命令，查看cuDNN的版本信息：

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

如果成功显示cuDNN的版本信息，则表示配置成功。

现在，您已经成功在Ubuntu上配置了cuDNN，可以开始在深度学习项目中使用它了。