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

Ubuntu系统NVIDIA显卡驱动及CUDA安装完全指南(小白也能看懂的详细教程)

Ubuntu系统NVIDIA显卡驱动及CUDA安装完全指南(小白也能看懂的详细教程)

本教程将详细介绍在Ubuntu系统上安装NVIDIA显卡驱动和CUDA的步骤,即使是初学者也能轻松跟随。正确安装NVIDIA驱动安装是确保GPU性能的关键,而CUDA安装教程则帮助您搭建深度学习环境。

前提条件

在开始之前,请确保您的系统是Ubuntu(如Ubuntu 20.04或更高版本),并且有一块NVIDIA显卡。本教程涵盖NVIDIA驱动安装和CUDA安装的基本步骤,适合新手。

步骤1:检查显卡和系统

打开终端,输入以下命令检查NVIDIA显卡型号:

    lspci | grep -i nvidia  

这将显示显卡信息。确保显卡支持CUDA。同时,更新Ubuntu系统:sudo apt update && sudo apt upgrade

步骤2:安装NVIDIA显卡驱动

在Ubuntu系统上,推荐使用PPA仓库进行NVIDIA驱动安装。首先,添加PPA:

    sudo add-apt-repository ppa:graphics-drivers/ppasudo apt update  

然后,安装适合您显卡的驱动版本。例如,安装驱动470:

    sudo apt install nvidia-driver-470  

安装完成后,重启系统:sudo reboot。重启后,运行nvidia-smi验证驱动。如果看到显卡信息,说明NVIDIA驱动安装成功。

Ubuntu系统NVIDIA显卡驱动及CUDA安装完全指南(小白也能看懂的详细教程) Ubuntu系统  NVIDIA驱动安装 CUDA安装教程 显卡驱动配置 第1张

步骤3:安装CUDA工具包

CUDA是NVIDIA的并行计算平台,对于深度学习等应用至关重要。这里提供CUDA安装教程。首先,访问NVIDIA官网下载CUDA,或使用命令行。以CUDA 11.4为例,下载安装包:

    wget https://developer.download.nvidia.com/compute/cuda/11.4.0/local_installers/cuda_11.4.0_470.42.01_linux.run  

运行安装脚本:

    sudo sh cuda_11.4.0_470.42.01_linux.run  

按照提示操作;在驱动安装选项中选择“否”(因为已安装)。安装完成后,配置环境变量。编辑~/.bashrc文件:nano ~/.bashrc,在末尾添加:

    export PATH=/usr/local/cuda-11.4/bin${PATH:+:${PATH}}export LD_LIBRARY_PATH=/usr/local/cuda-11.4/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}  

保存后运行source ~/.bashrc使配置生效。

步骤4:验证安装

验证CUDA安装:运行nvcc -V显示版本信息。同时,使用nvidia-smi检查驱动和CUDA兼容性。为了测试显卡驱动配置,可以运行CUDA示例程序。

结论

通过本教程,您应该已在Ubuntu系统上成功完成NVIDIA驱动安装和CUDA安装。正确的显卡驱动配置对于GPU加速计算和深度学习环境至关重要。如果遇到问题,参考官方文档或社区。保持驱动和CUDA更新以优化性能。