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


