当前位置:首页 > 系统教程 > 正文

Ubuntu 22.04安装CUDA 12.8完整指南

Ubuntu 22.04安装CUDA 12.8完整指南

一步步配置深度学习环境(小白教程)

本教程将详细指导你在Ubuntu 22.04系统上安装CUDA 12.8,以便进行深度学习和其他GPU加速计算。首先,确保安装正确的NVIDIA驱动,这是成功的关键。

Ubuntu 22.04安装CUDA 12.8完整指南 22.04  CUDA 12.8 NVIDIA驱动 深度学习 第1张

前提条件

  • 运行Ubuntu 22.04的计算机
  • NVIDIA GPU(支持CUDA)
  • 终端访问和sudo权限
  • 稳定互联网连接

步骤1:更新系统

打开终端,运行以下命令更新包列表和升级系统:

sudo apt updatesudo apt upgrade -y

这确保系统最新,避免兼容性问题。

步骤2:安装NVIDIA驱动

Ubuntu 22.04通常自带开源驱动,但为了CUDA,需要专有NVIDIA驱动。首先,检查可用驱动:

ubuntu-drivers devices

然后,安装推荐驱动(例如版本535):

sudo apt install nvidia-driver-535 -y

安装后重启:sudo reboot。重启后,运行nvidia-smi验证驱动安装。

步骤3:下载CUDA 12.8安装包

访问NVIDIA官网或使用wget下载CUDA 12.8。对于Ubuntu 22.04,使用以下命令:

wget https://developer.download.nvidia.com/compute/cuda/12.8.0/local_installers/cuda_12.8.0_550.54.15_linux.run

这下载CUDA 12.8安装程序到当前目录。

步骤4:安装CUDA 12.8

给予执行权限并运行安装程序:

chmod +x cuda_12.8.0_550.54.15_linux.runsudo ./cuda_12.8.0_550.54.15_linux.run

在安装界面中:

  1. 阅读并接受许可协议。
  2. 取消选中“Install NVIDIA Accelerated Graphics Driver”(如果已安装驱动)。
  3. 确保选中CUDA Toolkit 12.8,其他默认。
  4. 点击Install开始安装。
安装完成后,会显示摘要。

步骤5:配置环境变量

将CUDA路径添加到系统环境。编辑bash配置文件:

nano ~/.bashrc

在文件末尾添加:

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

保存退出(Ctrl+X,然后Y和Enter)。使更改生效:

source ~/.bashrc

步骤6:验证安装

检查CUDA版本确认安装成功:

nvcc --version

输出应显示CUDA 12.8。也可以编译示例测试:

cd /usr/local/cuda-12.8/samplessudo make./bin/x86_64/linux/release/deviceQuery

如果显示“Result = PASS”,则CUDA 12.8在Ubuntu 22.04上完全可用。

现在,你已成功安装CUDA 12.8,可以用于深度学习框架如TensorFlow或PyTorch。如有问题,检查驱动兼容性或重新安装。