Ubuntu系统NVIDIA显卡驱动及CUDA安装

1. 准备工作

在开始之前，请确认你的系统满足以下条件：

• Ubuntu系统（推荐20.04或22.04版本，关键词：Ubuntu 20.04）
• NVIDIA显卡（如GeForce、Quadro或Tesla系列，关键词：NVIDIA驱动）
• 稳定的网络连接

2. 安装NVIDIA驱动

有多种方法可以安装驱动，这里推荐最简单的方式：使用Ubuntu自带的“软件和更新”工具，或者通过命令行添加显卡驱动PPA。

首先，更新系统软件包：sudo apt update && sudo apt upgrade -y

然后，自动检测推荐驱动：ubuntu-drivers devices 会列出可用驱动，选择带有“recommended”标记的版本进行安装：sudo apt install nvidia-driver-470（版本号可能不同）。

安装完成后，重启系统：sudo reboot

重启后，运行 nvidia-smi 命令，如果看到显卡信息，说明驱动安装成功。

3. 安装CUDA工具包

CUDA是NVIDIA的并行计算平台，对于深度学习框架如TensorFlow、PyTorch至关重要（关键词：CUDA工具包）。推荐从NVIDIA官网下载runfile或deb包安装。

访问NVIDIA CUDA下载页面，选择你的系统版本（Linux -> x86_64 -> Ubuntu -> 20.04 -> runfile/local）。按照页面提示下载并安装。

例如，通过wget下载：wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run

然后运行安装：sudo sh cuda_11.8.0_520.61.05_linux.run，按照提示接受协议，取消安装驱动（因为已装），选择安装CUDA工具包即可。

安装后，需要配置环境变量。在 ~/.bashrc 末尾添加：

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

然后 source ~/.bashrc 使其生效。

4. 验证CUDA安装

运行 nvcc -V 查看CUDA编译器版本，或者运行 nvidia-smi 确认驱动和CUDA版本匹配。此时，你已经成功搭建了GPU环境，可以开始进行深度学习等任务（关键词：深度学习）。

5. 常见问题

如果安装驱动后无法进入图形界面，可能是驱动冲突，建议在命令行模式卸载原有驱动重装。更多问题可查阅NVIDIA官方文档。

© 2023 Ubuntu NVIDIA CUDA 安装教程