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

Ubuntu 22.04安装NVIDIA驱动与CUDA完整教程(一步步教你配置深度学习环境)

Ubuntu 22.04安装NVIDIA驱动与CUDA完整教程(一步步教你配置深度学习环境)

在Ubuntu 22.04系统中安装NVIDIA驱动和CUDA是进行机器学习和深度学习开发的关键步骤。本教程将详细指导小白用户完成整个过程,确保系统能识别GPU并加速计算。我们会从基础检查开始,逐步安装驱动和CUDA,最后验证安装成功。

一、准备工作:更新系统和检查硬件

首先,打开终端(Ctrl+Alt+T),更新系统包列表以确保获取最新软件版本。在终端中输入以下命令:

sudo apt update && sudo apt upgrade -y

完成后,检查您的NVIDIA显卡型号。使用命令 lspci | grep -i nvidia 来查看显卡信息。这将帮助您确认系统是否检测到NVIDIA GPU,为安装NVIDIA驱动做准备。

二、安装NVIDIA驱动

Ubuntu 22.04提供了多种安装NVIDIA驱动的方法。最简单的是通过“附加驱动”或终端。这里我们使用终端方式:

  1. 首先,添加Ubuntu官方显卡驱动仓库:
    sudo add-apt-repository ppa:graphics-drivers/ppa -y
  2. 更新包列表:
    sudo apt update
  3. 查看可用的NVIDIA驱动版本:
    ubuntu-drivers devices
  4. 安装推荐的NVIDIA驱动(例如版本525):
    sudo apt install nvidia-driver-525 -y

安装过程中,系统可能会提示禁用Secure Boot,根据提示操作即可。完成后,重启系统使NVIDIA驱动生效:sudo reboot

Ubuntu 22.04安装NVIDIA驱动与CUDA完整教程(一步步教你配置深度学习环境) 22.04  NVIDIA驱动 CUDA安装 深度学习环境 第1张

重启后,打开终端运行 nvidia-smi 命令,如果显示GPU信息,则表示NVIDIA驱动安装成功。这是配置CUDA安装的基础。

三、安装CUDA工具包

CUDA是NVIDIA的并行计算平台,需从官网下载。首先访问 NVIDIA CUDA下载页,选择Linux、x86_64、Ubuntu 22.04版本。这里我们使用网络安装方式:

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-ubuntu2204.pinsudo mv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600wget https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda-repo-ubuntu2204-12-2-local_12.2.0-535.54.03-1_amd64.debsudo dpkg -i cuda-repo-ubuntu2204-12-2-local_12.2.0-535.54.03-1_amd64.debsudo cp /var/cuda-repo-ubuntu2204-12-2-local/cuda-*-keyring.gpg /usr/share/keyrings/sudo apt updatesudo apt install cuda -y

安装完成后,需要设置环境变量以将CUDA添加到系统路径。打开终端,编辑bash配置文件:nano ~/.bashrc,在文件末尾添加以下行:

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

保存退出后,运行 source ~/.bashrc 使更改生效。现在,CUDA安装已完成,您可以开始使用它进行深度学习环境开发。

四、验证安装和常见问题

验证NVIDIA驱动和CUDA是否安装成功:

  • 运行 nvidia-smi:应显示GPU详细信息,包括驱动版本和CUDA版本。
  • 运行 nvcc --version:应显示CUDA编译器版本,确认CUDA安装正确。

如果遇到问题,如驱动冲突或CUDA路径错误,可尝试卸载重装。卸载NVIDIA驱动使用 sudo apt purge nvidia-*,卸载CUDA使用 sudo apt autoremove cuda。确保在Ubuntu 22.04系统中遵循步骤顺序,通常能成功配置。

本教程涵盖了从零开始安装NVIDIA驱动和CUDA的全过程,适合小白用户。通过正确配置,您的Ubuntu 22.04系统将准备好运行深度学习框架如TensorFlow或PyTorch。如有其他问题,请参考NVIDIA官方文档或社区论坛。