Ubuntu系统安装cuDNN详细步骤

Ubuntu系统安装cuDNN（CUDA Deep Neural Network Library）通常包含几个步骤，因为cuDNN是NVIDIA深度学习平台的一部分，需要先安装CUDA然后再安装cuDNN。以下是安装cuDNN的基本步骤：

1. **更新系统和安装依赖**

   sudo apt-get update
   sudo apt-get install -y build-essential cmake git libncurses5-dev pkg-config libprotobuf-dev libleveldb-dev libsnappy-dev liblz4-tool libzmq3-dev libhdf5-dev protobuf-compiler

2. **下载CUDA**
访问NVIDIA官网下载对应版本的CUDA SDK（https://developer.nvidia.com/cuda-downloads）。安装完成后，记得添加CUDA路径到环境变量。

3. **下载cuDNN**
登录NVIDIA Developer账户并访问cuDNN库页面（https://developer.nvidia.com/cudnn），选择对应的Ubuntu版本和CUDA版本进行下载。解压下载的cuDNN文件，并设置安装目录（例如`/usr/local/cuda`）。

4. **配置CUDA和cuDNN**
进入cuDNN源码目录，运行命令生成Makefile：

   export CUDA_HOME=/usr/local/cuda
   export CUDNN_INSTALL_DIR=$(pwd)
   mkdir build
   cd build
   cmake .. -DCMAKE_INSTALL_PREFIX=$CUDA_HOME -DCMAKE_BUILD_TYPE=Release

5. **编译和安装**

   make -j$(nproc) # 使用所有可用处理器进行并行编译
   sudo make install

6. **验证安装**

   CUDA_PATH=$CUDA_HOME
   export LD_LIBRARY_PATH=$CUDA_PATH/lib64:$LD_LIBRARY_PATH
   echo $LD_LIBRARY_PATH
   cudnnTestConvolution -printVersion