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

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

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

本教程面向 Ubuntu 24.04 用户,手把手教你完成 CUDA安装,并配置好 NVIDIA驱动 和基础的 深度学习环境。无论你是AI初学者还是迁移新系统的开发者,都能轻松跟随。

📋 准备工作

  • ✅ 一台已安装 Ubuntu 24.04 的电脑(物理机或云服务器)
  • ✅ 拥有NVIDIA独立显卡(本文以RTX 3060为例)
  • ✅ 稳定的网络连接(用于下载驱动和CUDA)
  • ✅ sudo权限的用户账户
Ubuntu 24.04 CUDA完整安装指南(从零搭建深度学习环境)  CUDA安装 NVIDIA驱动 深度学习环境 第1张

▲ CUDA Toolkit 与 NVIDIA 驱动关系示意

1️⃣ 更新系统并安装必要工具

打开终端(Ctrl+Alt+T),依次执行:

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

这些工具将帮助后续编译和安装 NVIDIA驱动

2️⃣ 卸载旧版NVIDIA驱动(如有)

如果系统曾经安装过NVIDIA驱动,建议彻底卸载:

sudo apt purge nvidia* -ysudo apt autoremove -ysudo apt autoclean

3️⃣ 安装NVIDIA驱动(推荐官方PPA或runfile)

Ubuntu 24.04 软件源已包含较新的驱动,但为获得最佳兼容性,建议添加显卡驱动PPA:

sudo add-apt-repository ppa:graphics-drivers/ppa -ysudo apt updatesudo apt install nvidia-driver-550 -y   # 550是目前稳定版本,可按需更换

安装完成后必须重启sudo reboot

重启后运行 nvidia-smi,若显示GPU信息,则驱动安装成功。

4️⃣ 选择CUDA版本并下载

访问 NVIDIA CUDA Toolkit 下载页,选择 Linux → x86_64 → Ubuntu → 24.04 → runfile(local)。本文以 CUDA 12.4 为例:

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

重要:执行runfile时,取消勾选“Driver”(因为驱动已单独安装),仅选择CUDA Toolkit并接受条款。

5️⃣ 配置环境变量

编辑 ~/.bashrc:

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

若你安装其他CUDA版本,请相应修改路径。

6️⃣ 验证CUDA安装

运行:

nvcc -V

若显示CUDA版本信息,则CUDA安装成功。你已拥有完整的深度学习环境基础。

🎯 进阶:安装cuDNN(加速神经网络训练)

从NVIDIA官网下载cuDNN(需要注册),解压后复制到CUDA目录:

tar -xvf cudnn-linux-x86_64-8.x.x.x_cudaX.Y-archive.tar.xzsudo cp cudnn--archive/include/cudnn.h /usr/local/cuda/include/sudo cp -P cudnn--archive/lib/libcudnn /usr/local/cuda/lib64/sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

❓ 常见问题

  • nvidia-smi 找不到命令? 驱动未正确安装,尝试重新安装或检查secure boot。
  • nvcc -V 提示未找到? 环境变量未配置或CUDA路径错误。
  • CUDA版本与驱动不兼容? 使用 nvidia-smi 顶部显示的CUDA版本作为上限,选择匹配的CUDA Toolkit。

—— 至此,你在 Ubuntu 24.04 上的 CUDA安装 圆满结束,尽情享受 深度学习 的乐趣吧!