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

Ubuntu 22.04安装CUDA Toolkit 12.4完全指南 (从零开始配置深度学习环境)

Ubuntu 22.04安装CUDA Toolkit 12.4完全指南 (从零开始配置深度学习环境)

本文将详细介绍在 Ubuntu 22.04 系统上安装 CUDA Toolkit 12.4 的完整步骤,帮助您快速搭建 深度学习环境。无论您是AI初学者还是经验丰富的开发者,按照本指南操作即可顺利完成 NVIDIA驱动安装 和CUDA配置。

Ubuntu 22.04安装CUDA Toolkit 12.4完全指南 (从零开始配置深度学习环境) CUDA 12.4  深度学习环境配置 NVIDIA驱动安装 第1张

图:验证CUDA安装成功的终端输出

1. 系统准备与依赖检查

在开始安装之前,请确保您的系统满足以下条件:

  • 操作系统:Ubuntu 22.04 LTS(64位)
  • 已安装NVIDIA显卡,并确认显卡支持CUDA 12.4(查看官方兼容列表
  • 拥有sudo权限的用户账户
  • 稳定的网络连接(用于下载安装包)

2. 安装NVIDIA驱动程序

CUDA Toolkit需要与NVIDIA驱动配合使用。推荐使用系统自带的apt仓库安装官方驱动:

sudo apt updatesudo apt upgradeubuntu-drivers devices   # 查看推荐驱动版本sudo apt install nvidia-driver-535   # 选择推荐版本(如535)sudo reboot

重启后运行 nvidia-smi 验证驱动是否安装成功。如果看到GPU信息,说明 NVIDIA驱动安装 完成。

3. 安装CUDA Toolkit 12.4

我们使用NVIDIA官方提供的.run文件进行安装,以确保最大兼容性。访问 CUDA Toolkit 12.4 下载页面,选择Linux -> x86_64 -> Ubuntu -> 22.04 -> runfile,然后执行以下命令:

wget https://developer.download.nvidia.com/compute/cuda/12.4.0/local_installers/cuda_12.4.0_550.54.15_linux.runsudo sh cuda_12.4.0_550.54.15_linux.run

在安装界面中,取消勾选“Driver”(因为我们已经单独安装了驱动),只保留CUDA Toolkit和CUDA Samples等组件。安装完成后,会提示将CUDA添加到PATH。

4. 配置环境变量

编辑 ~/.bashrc 文件,在末尾添加以下内容:

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

然后执行 source ~/.bashrc 使配置生效。

5. 验证CUDA安装

运行以下命令检查CUDA版本和编译器:

nvcc --versionnvidia-smi

如果两个命令都显示CUDA 12.4相关信息,说明 CUDA Toolkit 12.4 已成功安装,您的 深度学习环境配置 已完成!

🎉 恭喜!现在您可以在Ubuntu 22.04上使用CUDA 12.4进行深度学习了。接下来可以安装cuDNN、TensorFlow或PyTorch等框架。

本文关键词:CUDA Toolkit 12.4Ubuntu 22.04安装CUDA深度学习环境配置NVIDIA驱动安装。希望本教程对您有所帮助!