ubuntu设置cudnn

在Ubuntu上安装CUDA和cuDNN通常需要几个步骤，因为cuDNN是专为CUDA设计的深度学习库。以下是基本的安装过程：

1. **更新系统**:

   sudo apt-get update
   sudo apt-get upgrade

2. **安装CUDA**:
- 访问NVIDIA官网下载适用于Ubuntu的CUDA版本，安装时可能需要添加NVIDIA官方的PPA (Personal Package Archive)。
- 安装命令示例：

     sudo apt-add-repository ppa:graphics-drivers/ppa
     sudo apt-get install cuda toolkit

3. **安装cuDNN**:
- 下载cuDNN对应CUDA版本的.tgz文件，并解压到合适的路径（如`/usr/local/cuda/extras/CUDDN`）。
- 需要手动配置环境变量和路径，例如添加`lib64`目录到LD_LIBRARY_PATH：

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

4. **验证安装**:
- 运行`nvidia-smi`检查GPU是否正常识别，并通过`cd /path/to/cuDNN; make`来确认cuDNN是否安装成功。

5. **安装PyTorch或其他深度学习框架**时，记得指定cuDNN路径，例如对于PyTorch：

    pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/torch_stable.html --index-url https://pypi.org/simple --ldflags="-L/usr/local/cuda/lib64/"