当前位置:首页 > 系统教程 > 正文

Ubuntu 22.04 安装Nvidia驱动、CUDA 12.0与cuDNN 8.9.7终极指南(史上最详细小白友好教程)

Ubuntu 22.04 安装Nvidia驱动、CUDA 12.0与cuDNN 8.9.7终极指南(史上最详细小白友好教程)

本教程将详细介绍在Ubuntu 22.04系统上安装Nvidia驱动、CUDA 12.0和cuDNN 8.9.7的全过程,适用于深度学习、AI开发等场景。步骤清晰,Ubuntu 22.04用户即使无经验也能轻松跟随。

Ubuntu 22.04 安装Nvidia驱动、CUDA 12.0与cuDNN 8.9.7终极指南(史上最详细小白友好教程)  Nvidia驱动 CUDA 12.0 cuDNN 8.9.7 第1张

一、准备工作

在开始安装前,请确保系统已更新。打开终端(Ctrl+Alt+T),运行以下命令:

  • sudo apt update – 更新软件包列表。
  • sudo apt upgrade -y – 升级所有软件包。
  • sudo apt autoremove -y – 清理无用包。

检查系统是否已安装旧版Nvidia驱动,若有请先卸载:sudo apt purge nvidia*

二、安装Nvidia驱动

推荐使用Ubuntu仓库安装稳定版驱动,兼容性好。步骤如下:

  1. 添加PPA仓库:sudo add-apt-repository ppa:graphics-drivers/ppa -y
  2. 更新并安装驱动:sudo apt update,然后sudo apt install nvidia-driver-535 -y(535为示例版本,可根据显卡型号调整)。
  3. 重启系统:sudo reboot
  4. 验证安装:运行nvidia-smi,应显示GPU信息。这表明Nvidia驱动已成功安装。

三、安装CUDA 12.0

CUDA 12.0是Nvidia的并行计算平台,需从官网下载。选择runfile安装方式,更灵活:

  1. 访问Nvidia官网下载CUDA 12.0 runfile文件(如cuda_12.0.0_linux.run)。
  2. 在终端中进入下载目录,运行:sudo sh cuda_12.0.0_linux.run
  3. 安装过程中,按提示操作:接受协议,不安装驱动(因已安装),其他选项默认。
  4. 添加环境变量:编辑~/.bashrc文件,添加以下行:
    • export PATH=/usr/local/cuda-12.0/bin:$PATH
    • export LD_LIBRARY_PATH=/usr/local/cuda-12.0/lib64:$LD_LIBRARY_PATH
  5. 使配置生效:source ~/.bashrc
  6. 验证CUDA 12.0:运行nvcc -V,应显示版本信息。

四、安装cuDNN 8.9.7

cuDNN 8.9.7是深度学习库,需Nvidia开发者账号下载。安装步骤:

  1. 从Nvidia开发者网站下载cuDNN 8.9.7 for CUDA 12.0的压缩包(如cudnn-12.0-linux-x64-v8.9.7.29.tgz)。
  2. 解压并复制文件:
    • tar -xzvf cudnn-12.0-linux-x64-v8.9.7.29.tgz
    • sudo cp cuda/include/cudnn*.h /usr/local/cuda-12.0/include
    • sudo cp cuda/lib64/libcudnn* /usr/local/cuda-12.0/lib64
    • sudo chmod a+r /usr/local/cuda-12.0/include/cudnn*.h /usr/local/cuda-12.0/lib64/libcudnn*
  3. 验证cuDNN 8.9.7:运行cat /usr/local/cuda-12.0/include/cudnn_version.h | grep CUDNN_MAJOR -A 2,应显示版本号。

五、验证完整安装

运行以下命令确保所有组件正常工作:

  • nvidia-smi – 显示GPU和驱动信息。
  • nvcc -V – 检查CUDA 12.0版本。
  • cat /usr/local/cuda-12.0/include/cudnn_version.h | grep CUDNN_MAJOR – 确认cuDNN 8.9.7。

若所有命令输出正常,则安装成功!您现在可以在Ubuntu 22.04上使用Nvidia GPU进行深度学习了。

六、常见问题与总结

问题1:驱动安装失败?尝试禁用nouveau驱动:编辑/etc/modprobe.d/blacklist.conf,添加blacklist nouveau,然后sudo update-initramfs -u并重启。

问题2:CUDA路径错误?确保环境变量设置正确,并重启终端。

本教程覆盖了从Nvidia驱动CUDA 12.0cuDNN 8.9.7的全流程,强调步骤细节,适合初学者。如有疑问,请参考官方文档或社区论坛。