ubuntu安装cuDNN

首先，cuDNN是NVIDIA专门为深度学习框架提供的加速库，它可以优化卷积神经网络的训练和推理过程。如果你想在Ubuntu上使用cuDNN，需要安装CUDA和cuDNN两个软件包。

下面是具体的安装步骤：

1. 安装CUDA
首先，你需要先安装CUDA，可以从NVIDIA的官网上下载对应版本的CUDA安装包。下载完成后，打开终端，进入到安装包所在的目录，执行以下命令进行安装：

   sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb
   sudo apt-get update
   sudo apt-get install cuda

2. 安装cuDNN
下载对应版本的cuDNN安装包，需要注意的是，cuDNN需要与已安装的CUDA版本对应。下载完成后，解压缩文件，执行以下命令进行安装：

   sudo cp cuda/include/cudnn.h /usr/local/cuda/include
   sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
   sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

安装完成后，可以通过以下命令检查cuDNN是否正确安装：

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

如果输出结果中显示了CUDA和cuDNN的版本信息，则说明安装成功。

3. 配置环境变量
在终端中打开.bashrc文件，在文件末尾添加以下两行：

   export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
   export PATH=/usr/local/cuda/bin:$PATH

至此，cuDNN的安装和配置就完成了。如果你想在深度学习框架中使用cuDNN加速，只需要将对应的库链接到项目中即可。