Ubuntu 22.04 安装 CUDA 12.2

Ubuntu 22.04 安装 CUDA 12.2 完整指南

前言

CUDA 是 NVIDIA 推出的通用并行计算架构,广泛应用于深度学习、科学计算等领域。本指南将详细介绍在 Ubuntu 22.04 系统上安装 CUDA 12.2 的完整过程,包括驱动安装、环境配置和验证步骤。

系统要求

  • 操作系统:Ubuntu 22.04 LTS
  • 硬件要求:NVIDIA GPU(支持 CUDA)
  • 网络连接:用于下载安装包
  • 管理员权限:需要 sudo 权限

第一步:准备工作

  1. 更新系统

    sudo apt update && sudo apt upgrade -y
    
  2. 安装必要工具

    sudo apt install -y build-essential dkms gcc-12 g++-12
    
  3. 清理旧驱动(如有)

    sudo apt purge '^nvidia.*'
    sudo apt autoremove --purge
    sudo rm -rf /var/lib/dkms/nvidia
    

第二步:安装 NVIDIA 驱动

  1. 检查可用驱动版本

    ubuntu-drivers devices
    
  2. 安装推荐驱动(例如:535版本)

    sudo apt install -y nvidia-driver-535
    
  3. 重启系统

    sudo reboot
    
  4. 验证驱动安装

    nvidia-smi
    

    输出应显示类似以下内容:

    NVIDIA-SMI 535.86.05    Driver Version: 535.86.05    CUDA Version: 12.2
    

第三步:安装 CUDA Toolkit 12.2

  1. 下载 CUDA 安装包

    wget https://developer.download.nvidia.com/compute/cuda/12.2.2/local_installers/cuda_12.2.2_535.104.05_linux.run
    
  2. 运行安装程序

    sudo sh cuda_12.2.2_535.104.05_linux.run --silent --toolkit
    
  3. 配置环境变量

    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
    

第四步:验证安装

  1. 检查 CUDA 编译器

    nvcc -V
    

    输出应类似于:

    nvcc: NVIDIA (R) Cuda compiler
    release 12.2, V12.2.128
    
  2. 运行示例程序

    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 官方文档和示例代码。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值