跳到主要内容
极客日志极客日志面向AI+效率的开发者社区
首页博客GitHub 精选镜像工具UI配色美学隐私政策关于联系
搜索内容 / 工具 / 仓库 / 镜像...⌘K搜索
注册
博客列表
Shell / BashAI算法

ubuntu 24.04 安装 Nvidia 显卡驱动 + CUDA + cuDNN,配置 AI 深度学习训练环境

介绍在 ubuntu 24.04 环境下配置 Nvidia 显卡驱动、CUDA 及 cuDNN 以构建 AI 深度学习训练环境的完整流程。主要步骤包括使用 lspci 查询显卡型号,通过官网下载并编译安装对应版本的显卡驱动(需注意 gcc 版本兼容性),根据 nvidia-smi 支持的 CUDA 版本下载安装包并配置环境变量,最后通过本地仓库安装适配的 cuDNN 库。过程中涉及禁用 nouveau 驱动、确认各组件版本匹配关系及环境变量设置,确保深度学习框架能正常调用 GPU 资源。

苹果系统发布于 2026/2/6更新于 2026/5/2833 浏览
ubuntu 24.04 安装 Nvidia 显卡驱动 + CUDA + cuDNN,配置 AI 深度学习训练环境

ubuntu 24.04 安装 Nvidia 显卡驱动 + CUDA + cuDNN,配置 AI 深度学习训练环境

1. 查看本机显卡型号

lspci | grep -i nvidia

输出示例:

01:00.0 3D controller: NVIDIA Corporation GM108M [GeForce 920MX] (rev a2)

其中 GeForce 920MX 为显卡型号。

2. 下载 Nvidia 显卡驱动

访问官网下载页面,根据显卡型号搜索并下载最新版驱动。

显卡驱动下载界面

选择查找后下载对应版本。

驱动搜索结果

3. 安装显卡驱动

安装编译环境

sudo apt update
sudo apt install gcc make

运行安装程序

chmod +x NVIDIA-Linux-x86_64-560.31.02.run
sudo ./NVIDIA-Linux-x86_64-560.31.02.run

选择 Continue installation 并回车。

安装过程

进入编译中...

编译过程

注意:编译时可能出现错误中断,原因是默认 gcc 版本(如 11)低于驱动所需版本(如 12)。需安装 gcc-12 并调整链接。

sudo apt install gcc-12
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/x86_64-linux-gnu-gcc-12 20

重新运行安装程序:

sudo ./NVIDIA-Linux-x86_64-560.31.02.run

编译通过后,选择 Yes。

确认安装

安装过程中会提示是否禁用 nouveau 驱动,选择是,NVIDIA 会自动屏蔽。若无特殊要求,其他选项均选是。

nouveau 设置

终端运行 nvidia-smi 查看版本号和显存。

nvidia-smi 输出

安装完成,重启系统。

4. CUDA 安装

查看显卡支持的 CUDA 版本

nvidia-smi

输出显示最高可用 CUDA 版本为 12.6。

CUDA 支持版本

到官网下载 CUDA

访问 CUDA Toolkit Archive 页面,注册登录后下载。

CUDA 下载页

选择 12.6 以下版本(如 12.2),根据系统版本选择相应项。

版本选择

获取下载地址和命令,以 CUDA Toolkit 12.2.2 为例:

wget https://developer.download.nvidia.com/compute/cuda/12.2.2/local_installers/cuda_12.2.2_535.104.05_linux.run
sudo sh cuda_12.2.2_535.104.05_linux.run

执行安装

chmod +x cuda_12.2.2_535.104.05_linux.run
sudo ./cuda_12.2.2_535.104.05_linux.run

选择 Continue,输入 accept,取消 Driver 选项(已安装驱动),选择 Install。

CUDA 安装选项

等待安装完成。

配置环境

nano ~/.bashrc

在文件末尾添加以下内容:

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

重载配置:

source ~/.bashrc

检查安装

nvcc -V

输出版本号即表示成功。

nvcc 版本

5. 安装 cuDNN

官网下载

访问 cuDNN 下载页面,选择相应项生成下载命令。

cuDNN 下载页

示例命令:

wget https://developer.download.nvidia.com/compute/cudnn/9.3.0/local_installers/cudnn-local-repo-ubuntu2404-9.3.0_1.0-1_amd64.deb
sudo dpkg -i cudnn-local-repo-ubuntu2404-9.3.0_1.0-1_amd64.deb
sudo cp /var/cudnn-local-repo-ubuntu2404-9.3.0/cudnn-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cudnn

一般最新显卡驱动适配最新 cuDNN。若不适合当前 CUDA,可安装历史版本。cuDNN 支持矩阵可参考官方文档。

cuDNN 支持矩阵

查找 cuDNN 支持的 CUDA 历史版本,选择适应的版本下载 deb 包安装即可。

历史版本选择

目录

  1. ubuntu 24.04 安装 Nvidia 显卡驱动 + CUDA + cuDNN,配置 AI 深度学习训练环境
  2. 1. 查看本机显卡型号
  3. 2. 下载 Nvidia 显卡驱动
  4. 3. 安装显卡驱动
  5. 4. CUDA 安装
  6. 5. 安装 cuDNN
  • 💰 8折买阿里云服务器限时8折了解详情
  • Magick API 一键接入全球大模型注册送1000万token查看
  • 🤖 一键搭建Deepseek满血版了解详情
  • 一键打造专属AI 智能体了解详情
极客日志微信公众号二维码

微信扫一扫,关注极客日志

微信公众号「极客日志V2」,在微信中扫描左侧二维码关注。展示文案:极客日志V2 zeeklog

更多推荐文章

查看全部
  • 护网行动(HVV)解析:红蓝对抗机制与技术要求
  • Android 开发者与大模型技术:机遇、挑战与学习路径
  • C++ gflags 命令行参数解析与 spdlog 日志库实战
  • 阿里开源 32B 大模型 Qwen1.5 性能与实测分析
  • 优秀程序员的 7 个关键习惯
  • Ubuntu 安装 OpenClaw 的一键脚本、Docker 及源码方法
  • Stable Diffusion 使用 LoRA 模型生成逼真人物服饰教程
  • Git 推送报错“密码认证不支持”?切换到 SSH 密钥方案
  • SpringBoot 整合 Zookeeper 常见错误总结
  • LeetCode 982. 位运算三元组等于零
  • OpenClaw Linux 本地 AI 智能体部署指南
  • 前端请求后端 404/405/500 状态码排查与解决指南
  • Ollama Windows 安装与使用指南:本地运行 Llama 等模型
  • SQL 用户管理与权限控制
  • 相交链表 - 双指针解法
  • 前端状态管理方案对比:Redux、Zustand 与 Pinia
  • 行星减速器原理、计算公式及 C++ 实现
  • 前端地图基本操作控制:平移、缩放、旋转及样式切换
  • 计算机网络:WebSocket 如何实现全双工通信
  • OpenClaw Web 控制台使用全解析:可视化配置与监控

相关免费在线工具

  • 加密/解密文本

    使用加密算法(如AES、TripleDES、Rabbit或RC4)加密和解密文本明文。 在线工具,加密/解密文本在线工具,online

  • RSA密钥对生成器

    生成新的随机RSA私钥和公钥pem证书。 在线工具,RSA密钥对生成器在线工具,online

  • Mermaid 预览与可视化编辑

    基于 Mermaid.js 实时预览流程图、时序图等图表,支持源码编辑与即时渲染。 在线工具,Mermaid 预览与可视化编辑在线工具,online

  • 随机西班牙地址生成器

    随机生成西班牙地址(支持马德里、加泰罗尼亚、安达卢西亚、瓦伦西亚筛选),支持数量快捷选择、显示全部与下载。 在线工具,随机西班牙地址生成器在线工具,online

  • Gemini 图片去水印

    基于开源反向 Alpha 混合算法去除 Gemini/Nano Banana 图片水印,支持批量处理与下载。 在线工具,Gemini 图片去水印在线工具,online

  • Base64 字符串编码/解码

    将字符串编码和解码为其 Base64 格式表示形式即可。 在线工具,Base64 字符串编码/解码在线工具,online