ubuntu中cudnn安装

在 Ubuntu 中安装 CUDA 和 cuDNN，您可以按照以下步骤进行操作：

1. 安装 CUDA Toolkit：
- 前往 NVIDIA 官网下载适用于 Ubuntu 的 CUDA Toolkit 安装包。
- 打开终端，进入下载目录，执行以下命令安装 CUDA Toolkit：

     sudo dpkg -i cuda-repo-<version>.deb
     sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/<version>/x86_64/7fa2af80.pub
     sudo apt-get update
     sudo apt-get install cuda

- 安装完成后，将 CUDA 的 bin 和 lib 目录添加到 PATH 和 LD_LIBRARY_PATH 环境变量中。您可以编辑 `~/.bashrc` 文件，在文件末尾添加以下行，并执行 `source ~/.bashrc` 使配置生效：

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

2. 安装 cuDNN：
- 前往 NVIDIA 开发者网站下载适用于您的 CUDA 版本的 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*

完成上述步骤后，您应该已成功安装了 CUDA 和 cuDNN。可以通过执行 `nvcc --version` 和 `cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2` 来验证安装是否成功。