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

Ubuntu 22.04安装CUDA Toolkit 12.4全攻略

Ubuntu 22.04安装CUDA Toolkit 12.4全攻略

从零开始配置GPU环境,小白也能看懂

本教程将详细介绍在Ubuntu 22.04系统上安装CUDA Toolkit 12.4的完整步骤。无论你是深度学习新手还是经验丰富的开发者,都能按照本指南轻松完成配置,为GPU计算任务做好准备。

一、安装前准备

在开始安装CUDA Toolkit 12.4之前,请确保你的系统是Ubuntu 22.04,并已连接互联网。同时,你需要一块兼容的NVIDIA显卡,并安装最新的NVIDIA驱动。如果你尚未安装驱动,可以按照以下步骤操作:

  1. 打开终端(快捷键 Ctrl+Alt+T)。
  2. 运行 sudo apt update 更新软件包列表。
  3. 运行 sudo apt install nvidia-driver-535 安装推荐驱动(版本可能更新,请根据提示调整)。
  4. 重启系统使驱动生效。
Ubuntu 22.04安装CUDA Toolkit 12.4全攻略 22.04  CUDA 12.4 NVIDIA驱动 GPU计算 第1张

二、安装CUDA Toolkit 12.4

以下是安装CUDA Toolkit 12.4的具体步骤:

  1. 访问NVIDIA官网获取CUDA安装命令,或直接在终端中执行以下命令下载安装包:wget https://developer.download.nvidia.com/compute/cuda/12.4.0/local_installers/cuda_12.4.0_550.54.14_linux.run
  2. 给予安装文件执行权限:sudo chmod +x cuda_12.4.0_550.54.14_linux.run
  3. 运行安装程序,根据提示选择“安装”选项(建议使用默认设置):sudo ./cuda_12.4.0_550.54.14_linux.run
  4. 安装过程中,如果系统提示已存在NVIDIA驱动,请确保不取消驱动安装,以避免冲突。
  5. 等待安装完成,这可能需要几分钟时间。

三、配置环境变量

安装完成后,需要配置环境变量以使CUDA Toolkit 12.4生效:

  1. 打开终端,编辑bash配置文件:nano ~/.bashrc
  2. 在文件末尾添加以下行:export PATH=/usr/local/cuda-12.4/bin${PATH:+:${PATH}}export LD_LIBRARY_PATH=/usr/local/cuda-12.4/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
  3. 保存文件(快捷键 Ctrl+O,然后按 Enter),退出编辑器(快捷键 Ctrl+X)。
  4. 使配置立即生效:source ~/.bashrc

四、验证安装

为了确保CUDA Toolkit 12.4Ubuntu 22.04上成功安装,请运行以下验证命令:

  1. 检查CUDA版本:nvcc --version如果输出显示“Cuda compilation tools, release 12.4”,则表示安装成功。
  2. 测试GPU计算能力:运行示例程序(如果有),或使用深度学习框架如TensorFlow或PyTorch进行简单测试。

五、常见问题与解决

如果在安装过程中遇到问题,请参考以下建议:

  • 驱动冲突:如果系统已有旧版驱动,请先使用 sudo apt purge nvidia-* 清除后再安装。
  • 路径错误:确保环境变量配置正确,可通过 echo $PATH 检查。
  • 权限问题:在命令前加 sudo 以获得管理员权限。

通过本教程,你应该已经成功在Ubuntu 22.04上安装了CUDA Toolkit 12.4,为后续的GPU计算和深度学习项目奠定了基础。如果有其他问题,欢迎查阅NVIDIA官方文档或社区论坛。