Ubuntu 22.04 安装 CUDA 12.2 完整指南
前言
CUDA 是 NVIDIA 推出的通用并行计算架构,广泛应用于深度学习、科学计算等领域。本指南将详细介绍在 Ubuntu 22.04 系统上安装 CUDA 12.2 的完整过程,包括驱动安装、环境配置和验证步骤。
系统要求
- 操作系统:Ubuntu 22.04 LTS
- 硬件要求:NVIDIA GPU(支持 CUDA)
- 网络连接:用于下载安装包
- 管理员权限:需要
sudo
权限
第一步:准备工作
-
更新系统
sudo apt update && sudo apt upgrade -y
-
安装必要工具
sudo apt install -y build-essential dkms gcc-12 g++-12
-
清理旧驱动(如有)
sudo apt purge '^nvidia.*' sudo apt autoremove --purge sudo rm -rf /var/lib/dkms/nvidia
第二步:安装 NVIDIA 驱动
-
检查可用驱动版本
ubuntu-drivers devices
-
安装推荐驱动(例如:535版本)
sudo apt install -y nvidia-driver-535
-
重启系统
sudo reboot
-
验证驱动安装
nvidia-smi
输出应显示类似以下内容:
NVIDIA-SMI 535.86.05 Driver Version: 535.86.05 CUDA Version: 12.2
第三步:安装 CUDA Toolkit 12.2
-
下载 CUDA 安装包
wget https://developer.download.nvidia.com/compute/cuda/12.2.2/local_installers/cuda_12.2.2_535.104.05_linux.run
-
运行安装程序
sudo sh cuda_12.2.2_535.104.05_linux.run --silent --toolkit
-
配置环境变量
echo 'export PATH=/usr/local/cuda-12.2/bin:$PATH' >> ~/.bashrc echo 'export LD_LIBRARY_PATH=/usr/local/cuda-12.2/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc source ~/.bashrc
第四步:验证安装
-
检查 CUDA 编译器
nvcc -V
输出应类似于:
nvcc: NVIDIA (R) Cuda compiler release 12.2, V12.2.128
-
运行示例程序
cd /usr/local/cuda-12.2/samples/1_Utilities/deviceQuery sudo make ./deviceQuery
成功输出示例:
deviceQuery, CUDA Driver = CUDARuntime, CUDA Driver Version = 12.2, CUDA Runtime Version = 12.2 ... Result = PASS
常见问题解决
1. 驱动安装失败
尝试以下命令修复:
sudo apt install -f
sudo dkms install -m nvidia -v 535.86.05
sudo reboot
2. CUDA 示例编译错误
确保已安装正确的 GCC 版本:
sudo apt install gcc-12 g++-12
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-12 100
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-12 100
3. 内核模块签名问题(Secure Boot)
禁用验证:
sudo mokutil --disable-validation
sudo reboot
卸载 CUDA
sudo /usr/local/cuda-12.2/bin/uninstall_cuda_12.2.pl
sudo apt purge '^nvidia.*'
sudo apt autoremove
结论
通过本指南,您已经成功在 Ubuntu 22.04 系统上安装了 CUDA 12.2 开发环境。现在可以开始使用 CUDA 加速您的计算任务。如需进一步学习 CUDA 编程,请参考 NVIDIA 官方文档和示例代码。