讲师中心 微信公众号
AI工具推荐 视频效率加速

怎么在docker-compose.yml中通过devices参数调用物理GPU

大丽酱_2505

大丽酱_2505

发布时间:2026-05-28 11:41:00

|

520人浏览过

|

来源于php中文网

原创

在 docker-compose.yml 中不能仅靠 devices 挂载 /dev/nvidia* 设备节点来使用 GPU,必须通过 gpus 字段(如 gpus: "all")配合 runtime: nvidia 和 NVIDIA Container Toolkit,由 nvidia-container-runtime 注入驱动、库及环境变量,才能真正启用 GPU 加速。

怎么在docker-compose.yml中通过devices参数调用物理gpu

在 docker-compose.yml 中通过 devices 参数直接调用物理 GPU 不可行,因为 devices 只能挂载设备文件(如 /dev/nvidia0),但不提供驱动、运行时库和 CUDA 上下文支持,容器内无法真正使用 GPU。

真正启用 GPU 的标准方式是使用 Docker 的 --gpus 机制(需 Docker 19.03+ 和 nvidia-container-toolkit),而 docker-compose 从 v2.20+(2022 年底起)开始原生支持 deploy.resources.reservations.devices,但更推荐、更通用、更可靠的方式是使用 runtime: nvidia + gpus(适用于 compose v2.17+)或 gpus under deploy(v2.20+)。下面分场景说明:


✅ 正确做法:用 gpus 字段(推荐)

确保满足前提:

  • 宿主机已安装 NVIDIA 驱动(nvidia-smi 可运行)
  • 已安装并配置 nvidia-container-toolkit
  • Docker 版本 ≥ 19.03,docker-compose ≥ v2.17(CLI 插件模式)
version: "3.8"
services:
  my-cuda-app:
    image: nvidia/cuda:12.2.0-runtime-ubuntu22.04
    runtime: nvidia  # 必须声明,否则 gpus 不生效(旧版必需,新版仍建议保留)
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1         # 或 "all"、"0,1" 等
              capabilities: [gpu, compute, utility]
    # 或更简洁写法(compose v2.20+ 支持的 shorthand):
    # gpus: "all"   # ← 这行等价于上面 deploy.devices 块(推荐!)

✅ gpus: "all" 是最常用写法;也可写 gpus: "0"、gpus: "0,1"、gpus: "device=0"。

验证是否生效:

Learn Docker
Learn Docker

通过 LabEx 容器基础课程和镜像/编排实验学习 Docker。Docker 课程和实验通过 LabEx MCP 获取,返回 labex.io 链接以继续。

下载
docker-compose run --rm my-cuda-app nvidia-smi

应正常显示 GPU 信息。


⚠️ 为什么不能只靠 devices:?

以下写法无效(即使容器能访问 /dev/nvidia*,也无法运行 CUDA 程序):

# ❌ 错误示范:仅挂载设备节点,无驱动/库/环境
devices:
  - "/dev/nvidia0:/dev/nvidia0"
  - "/dev/nvidiactl:/dev/nvidiactl"
  - "/dev/nvidia-uvm:/dev/nvidia-uvm"

原因:

  • 缺少 libcuda.so、libcudnn.so 等运行时库
  • 缺少 nvidia-smi、CUDA 工具链
  • 缺少 NVIDIA_VISIBLE_DEVICES、NVIDIA_DRIVER_CAPABILITIES 等关键环境变量
  • nvidia-container-runtime 才负责自动注入这些

? 补充:检查与排错

  • 确认 nvidia-container-toolkit 已注册为默认 runtime:
    cat /etc/docker/daemon.json
    # 应含:{"runtimes": {"nvidia": {"path": "nvidia-container-runtime", ...}}}
  • 重启 Docker:sudo systemctl restart docker
  • 查看可用 GPU:docker run --rm --gpus all nvidia/cuda:12.2.0-runtime-ubuntu22.04 nvidia-smi
  • Compose 版本检查:docker compose version(注意是 docker compose,不是 docker-compose)

? 小结关键点

  • devices: 用于挂载任意块设备(如 /dev/sdb),不适用于 GPU
  • GPU 必须用 gpus:(顶层字段)或 deploy.resources.reservations.devices(显式声明)
  • 必须配合 runtime: nvidia(旧版强制,新版建议保留)
  • 基础镜像需包含 CUDA runtime(如 nvidia/cuda:xx-runtime-ubuntuyy)
  • 宿主机驱动版本需兼容容器中 CUDA 版本(CUDA 兼容性表)

不复杂但容易忽略。

热门AI工具

更多
SkildArt
SkildArt Hot

SkildArt是一款AI文本写作工具,一站式 AI 视觉创作平台。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

超级简历WonderCV

一款AI办公效率工具,主要用于免费求职简历模版下载制作,应届生职场人必备简历制作神器,适合需要提升相关任务效率的用户。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

Laper
Laper Hot

Laper是专为编剧、导演和制片人推出的 AI 原生剧本创作工具。

讯飞绘文

讯飞绘文是一款由科大讯飞推出的一站式 AIGC 内容运营平台。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

VibeKnow
VibeKnow Hot

一款AI视频创作工具,主要用于全球首个AI知识视频创作平台,文档、文章、网页,一键生成视频,适合需要提升相关任务效率的用户。

蛙蛙写作

一款AI论文写作工具,主要用于超级AI智能写作助手,适合需要提升相关任务效率的用户。

相关专题

更多
C语言变量命名
C语言变量命名

c语言变量名规则是:1、变量名以英文字母开头;2、变量名中的字母是区分大小写的;3、变量名不能是关键字;4、变量名中不能包含空格、标点符号和类型说明符。php中文网还提供c语言变量的相关下载、相关课程等内容,供大家免费下载使用。

3289

2023.06.20

c语言入门自学零基础
c语言入门自学零基础

C语言是当代人学习及生活中的必备基础知识,应用十分广泛,本专题为大家c语言入门自学零基础的相关文章,以及相关课程,感兴趣的朋友千万不要错过了。

2348

2023.07.25

c语言运算符的优先级顺序
c语言运算符的优先级顺序

c语言运算符的优先级顺序是括号运算符 > 一元运算符 > 算术运算符 > 移位运算符 > 关系运算符 > 位运算符 > 逻辑运算符 > 赋值运算符 > 逗号运算符。本专题为大家提供c语言运算符相关的各种文章、以及下载和课程。

1280

2023.08.02

c语言数据结构
c语言数据结构

数据结构是指将数据按照一定的方式组织和存储的方法。它是计算机科学中的重要概念,用来描述和解决实际问题中的数据组织和处理问题。数据结构可以分为线性结构和非线性结构。线性结构包括数组、链表、堆栈和队列等,而非线性结构包括树和图等。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

1238

2023.08.09

c语言random函数用法
c语言random函数用法

c语言random函数用法:1、random.random,随机生成(0,1)之间的浮点数;2、random.randint,随机生成在范围之内的整数,两个参数分别表示上限和下限;3、random.randrange,在指定范围内,按指定基数递增的集合中获得一个随机数;4、random.choice,从序列中随机抽选一个数;5、random.shuffle,随机排序。

1376

2023.09.05

c语言const用法
c语言const用法

const是关键字,可以用于声明常量、函数参数中的const修饰符、const修饰函数返回值、const修饰指针。详细介绍:1、声明常量,const关键字可用于声明常量,常量的值在程序运行期间不可修改,常量可以是基本数据类型,如整数、浮点数、字符等,也可是自定义的数据类型;2、函数参数中的const修饰符,const关键字可用于函数的参数中,表示该参数在函数内部不可修改等等。

2218

2023.09.20

c语言get函数的用法
c语言get函数的用法

get函数是一个用于从输入流中获取字符的函数。可以从键盘、文件或其他输入设备中读取字符,并将其存储在指定的变量中。本文介绍了get函数的用法以及一些相关的注意事项。希望这篇文章能够帮助你更好地理解和使用get函数 。

3580

2023.09.20

c数组初始化的方法
c数组初始化的方法

c语言数组初始化的方法有直接赋值法、不完全初始化法、省略数组长度法和二维数组初始化法。详细介绍:1、直接赋值法,这种方法可以直接将数组的值进行初始化;2、不完全初始化法,。这种方法可以在一定程度上节省内存空间;3、省略数组长度法,这种方法可以让编译器自动计算数组的长度;4、二维数组初始化法等等。

16495

2023.09.22

Playwright 测试失败怎么办
Playwright 测试失败怎么办

Playwright测试失败专题汇集超时、定位器匹配、浏览器提前关闭、依赖与插件配置、文件上传、SSL及Docker启动等故障文章,并介绍通过截图、报告和Trace定位错误。

0

2026.10.21

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
phpMyAdmin 安装文档
phpMyAdmin 安装文档

共0课时 | 0人学习

Docker 官方文档
Docker 官方文档

共0课时 | 0人学习

宝塔 Docker 模块使用手册
宝塔 Docker 模块使用手册

共0课时 | 0人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn