Ubuntu 下 CUDA 环境安装与配置
在 Ubuntu 上配置 CUDA 环境，通常需要以下几个步骤：
- 检查系统需求和安装前的准备
- 安装 NVIDIA 驱动
- 安装 CUDA 工具包
- 配置环境变量
- 验证 CUDA 安装
- （可选）安装 cuDNN（用于深度学习）
1. 检查系统需求和准备工作
在安装 CUDA 之前，确保以下事项：
- 你使用的 Ubuntu 版本支持 CUDA。可以查看 NVIDIA CUDA 官方文档 获取兼容的操作系统列表。
- NVIDIA 显卡支持 CUDA（一般来说，较新的 NVIDIA 显卡都支持）。
建议你先删除任何旧版本的驱动程序或 CUDA 软件包，以避免冲突：
sudo apt-get --purge remove "*nvidia*"
2. 安装 NVIDIA 驱动
你可以使用 Ubuntu 的包管理器来自动安装适合的 NVIDIA 驱动。
首先，更新系统软件包：
sudo apt update && sudo apt upgrade
启用 Ubuntu 的驱动程序仓库：
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update
检查可用的 NVIDIA 驱动：
ubuntu-drivers devices
这将列出系统支持的 NVIDIA 驱动版本。安装推荐的 NVIDIA 驱动程序（例如 nvidia-driver-xxx，其中 xxx 是版本号）：
sudo apt install nvidia-driver-470
安装完成后，重新启动计算机：
sudo reboot
3. 安装 CUDA 工具包
下载 CUDA 工具包
访问 NVIDIA CUDA Toolkit 下载页面，选择你的 Ubuntu 版本并复制对应的安装命令。以 CUDA 12.0 为例，使用以下步骤：
添加 CUDA 存储库和 GPG 密钥
根据官网提供的命令，运行以下命令来添加 CUDA 存储库：
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
