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

Ubuntu 24.04 安装 CUDA 完全指南(从零搭建深度学习环境)

Ubuntu 24.04 安装 CUDA 完全指南(从零搭建深度学习环境)

欢迎来到这篇详细的 Ubuntu 24.04 CUDA 安装 教程!无论你是深度学习新手,还是希望升级环境的开发者,本文都会手把手带你完成每一步。CUDA 是 NVIDIA 的并行计算平台,对于运行 TensorFlow、PyTorch 等框架至关重要。通过本文,你将学会如何正确安装 NVIDIA 驱动、配置 CUDA 环境,最终成功搭建起高效的 深度学习环境搭建 基础。

Ubuntu 24.04 安装 CUDA 完全指南(从零搭建深度学习环境) CUDA安装 NVIDIA驱动安装 CUDA环境配置 深度学习环境搭建 第1张

1. 准备工作:检查硬件与系统更新

在开始 Ubuntu 24.04 CUDA 安装 前,请确保你的电脑配备了 NVIDIA 显卡。打开终端(Ctrl+Alt+T),输入以下命令检查显卡型号:

lspci | grep -i nvidia

如果看到 NVIDIA 设备信息,说明显卡已就绪。接着更新系统软件包,并安装必要的编译工具:

sudo apt update && sudo apt upgrade -ysudo apt install build-essential dkms linux-headers-$(uname -r) -y

2. 安装 NVIDIA 驱动

NVIDIA 驱动安装 是 CUDA 正常工作的前提。Ubuntu 24.04 提供了多种安装方式,推荐使用官方 PPA 源安装最新稳定版:

sudo add-apt-repository ppa:graphics-drivers/ppa -ysudo apt updateubuntu-drivers devices

查看推荐版本(如 nvidia-driver-550),然后安装:

sudo apt install nvidia-driver-550 -y

安装完成后重启系统:sudo reboot。重启后验证驱动:

nvidia-smi

如果显示 GPU 信息和驱动版本,说明驱动安装成功。

3. 下载 CUDA Toolkit

访问 NVIDIA 官方 CUDA 下载页面,选择 Linux → x86_64 → Ubuntu → 24.04 → runfile(或 deb 包)。本文以 runfile 为例,因为它更灵活且易于后续管理。复制下载链接,在终端中执行(示例版本为 CUDA 12.5,请根据实际情况替换):

wget https://developer.download.nvidia.com/compute/cuda/12.5.0/local_installers/cuda_12.5.0_555.42.02_linux.runsudo sh cuda_12.5.0_555.42.02_linux.run

注意:如果之前未安装驱动,安装器会询问是否安装驱动,由于我们已经单独安装了驱动,此处应取消勾选 Driver 选项,仅安装 CUDA Toolkit。

4. 配置 CUDA 环境变量

安装完成后,CUDA 默认安装在 /usr/local/cuda 下。接下来进行 CUDA 环境配置,编辑 ~/.bashrc 文件:

echo "export PATH=/usr/local/cuda/bin:$PATH" >> ~/.bashrcecho "export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH" >> ~/.bashrcsource ~/.bashrc

这样每次打开终端都会自动加载 CUDA 相关路径。

5. 验证 CUDA 安装

执行以下命令查看 CUDA 编译器版本:

nvcc -V

如果显示 CUDA 版本信息,则安装成功。进一步编译 CUDA 样例(可选):

cd /usr/local/cuda/samples/1_Utilities/deviceQuerysudo make./deviceQuery

若显示 Result = PASS,说明 CUDA 可以正常调用 GPU,至此 Ubuntu 24.04 CUDA 安装 圆满完成!

6. 常见问题与解决

  • 驱动冲突:如果安装 CUDA 时勾选了驱动,可能导致原有驱动异常,建议卸载旧驱动后重试。
  • 路径未生效:确保正确 source 了 .bashrc,或重启终端。
  • 缺少依赖:若编译样例时报错,请安装 libglu1-mesa-dev freeglut3-dev 等库。

结语

通过本教程,你不仅完成了 Ubuntu 24.04 CUDA 安装,还掌握了 NVIDIA 驱动安装CUDA 环境配置 的核心步骤。现在你可以放心地在系统中运行各类深度学习框架,享受 GPU 加速带来的高效体验。如果在安装过程中遇到任何问题,欢迎在评论区留言交流!