ubuntu安装 NVIDIA CUDA Toolkit 和 cuDNN 库

在Ubuntu系统上安装NVIDIA CUDA Toolkit和cuDNN库通常是为了支持深度学习框架如TensorFlow、PyTorch等的GPU加速计算。以下是安装步骤：

1. **安装CUDA Toolkit**:
- 更新软件包列表并安装依赖项：

     sudo apt-y curl gnupg lsb-release

- 添加NVIDIA官方GPG密钥：

     curl https://developer.nvidia.com苑公匙 | sudo apt-key add -
     echo "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64 ${CUDA_REPO_PKG_VERSION} main" | sudo tee /etc/apt/sources.list.d/cuda.list

- 确定你的CUDA版本（例如11.0）：

     export CUDA_VERSION=11.0

- 安装CUDA：

     sudo apt-get update
     sudo apt-get install cuda-cudart-$CUDA_VERSION -yy

- 可能还需要安装驱动程序，可以通过`nvidia-smi`检查。

2. **安装cuDNN**:
- 首先，从NVIDIA官网下载cuDNN压缩包，通常是.tgz文件。
- 解压到合适的目录，比如`/usr/local/cuda`：

     sudo mkdir -p /usr/local/cuda/include/usr/local/cuda/lib64
     tar -xvf cudnn*.tgz -C /usr/local/cuda
     rm cudnn*.tgz

- 设置环境变量指向cuDNN库：

     sudo cp -r /usr/local/cuda/include/* /usr/local/include/
     sudo cp -r /usr/local/cuda/lib64/* /usr/local/lib/

- 添加路径到LD_LIBRARY_PATH：

     echo 'export LD_LIBRARY_PATH=\$LD_LIBRARY_PATH:/usr/local/cuda/lib64' >> ~/.bashrc
     source ~/.bashrc