跳到主要内容
极客日志极客日志面向AI+效率的开发者社区
首页博客GitHub 精选镜像AI 生图工具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/7/2355 浏览
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
  • 免费图片AI生成工具免费生成了解详情
  • Magick API 一键接入全球大模型注册送1000万token查看
  • 免费图片视频在线生成30秒,将你的创意变成现实开始设计
  • X/Twitter免费视频下载器免登陆无限额度免费视频解析下载了解详情
  • 100+免费在线小游戏爽一把
极客日志微信公众号二维码

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

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

更多推荐文章

查看全部
  • LLM 推理思维链变体:TOT、GOT、AOT、SoT 与 PoT 详解
  • DeepSeek-R1 技术解析:训练流程与架构设计详解
  • UI UX Pro Max:让 AI 辅助生成专业级前端 UI 代码
  • C++ 递归入门:汉诺塔问题的思路与实现
  • 从 XMLHttpRequest 到 Fetch API:前端网络请求演进与迁移指南
  • C++ 内存管理解析:现代视角下的核心机制与实践
  • 三款主流云电脑部署 DeepSeek 模型性能实测与对比
  • JavaScript 基础语法中篇:运算符、条件、循环与数组详解
  • Windows 下使用 VSCode 结合 WSL 实现远程开发及公网访问配置
  • DeepSeek R1 与 Janus-Pro 本地部署高级教程
  • ComfyUI 核心架构与 AI 绘画工作流实战
  • 使用 Playwright 封装 Web 爬虫并隐藏自动化特征
  • 基于 Java 的校园二手物品在线交易平台设计与实现
  • QClaw 接入微信:AI 正从“会聊天”进化为“会干活”
  • 从冯诺依曼体系结构到进程管理详解
  • 使用 Python 将 CSV 数据导入 Neo4j 图数据库
  • 达梦数据库 JAVA 外部函数使用指南
  • 服务器端 VS Code Copilot 加载超时排查与优化指南
  • JavaScript 数组模拟栈与队列数据结构
  • Python 安全有效地处理配置的最佳实践

相关免费在线工具

  • 加密/解密文本

    使用加密算法(如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