【docker】docker下如何使用宿主主机的GPU

不能直接用,默认隔离;要装 NVIDIA 容器工具包并加参数才能用宿主机 GPU。


一、为什么默认不能用

Docker 容器默认用命名空间 + cgroups隔离:

  • 看不见 /dev/nvidia* 设备
  • 没有宿主机 CUDA 库和驱动
  • 必须靠 NVIDIA Container Toolkit 打通

二、前提(NVIDIA GPU)

  1. 宿主机装好 NVIDIA 驱动nvidia-smi 能跑)
  2. 安装 nvidia-container-toolkit(旧叫 nvidia-docker2)
  3. Docker ≥ 19.03 (支持 --gpus
Ubuntu 安装示例
bash 复制代码
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list

sudo apt update
sudo apt install -y nvidia-docker2
sudo systemctl restart docker

三、启动容器(关键参数)

--gpus 分配 GPU:

bash 复制代码
# 所有 GPU
docker run --gpus all -it nvidia/cuda:12.0-base nvidia-smi

# 只给 GPU 0
docker run --gpus device=0 -it nvidia/cuda:12.0-base nvidia-smi

# 给 GPU 0 和 1
docker run --gpus device=0,1 -it nvidia/cuda:12.0-base nvidia-smi

四、容器里能用到什么

  • 直接用宿主机物理 GPU(性能接近原生)
  • 容器内只需装应用 + CUDA 运行时,不用装驱动
  • 支持 CUDA、cuDNN、TensorRT、FFmpeg 硬编解码

五、你的场景(WebRTC / 音视频)

  • 如果是 NVIDIA GPU(如 Quadro、RTX)
    • 装好驱动 + 工具包,加 --gpus all
    • 容器内可直接用 NVENC/NVDEC 硬编硬解,降低 CPU 占用
  • 如果是 i5-7400 核显(HD 630)
    • Docker 对 Intel 核显支持有限,一般用 VA-API 直通
    • 启动加 --device /dev/dri,并挂载相关设备

六、一句话总结

默认隔离不能用;装 NVIDIA 工具包、用 --gpus 参数,容器就能直接调用宿主机 GPU 做硬编解码/AI 加速。

需要我给你一份可直接复制的安装和验证脚本吗?

结论先说:

Docker 18.09.7 不支持 --gpus 参数!

这个版本太老了,无法直接使用你刚才看到的现代 GPU 调用方式。


为什么不行?

  • --gpus all 这个参数是 Docker 19.03 版本才正式加入
  • 你的 18.09.7 属于旧版本,没有这个功能

那你现在有 2 个方案(都能用)

方案1:升级 Docker(推荐,最简单)

直接把 Docker 升级到 19.03 以上 (现在都 26+ 了)

升级后一句 --gpus all 就能用 GPU,最省心。


方案2:不升级 Docker,用旧版 nvidia-docker2(兼容 18.09)

你的 18.09.7 可以用 nvidia-docker2,只是命令不一样:

1. 安装 nvidia-docker2(和刚才一样)
bash 复制代码
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list

sudo apt update
sudo apt install -y nvidia-docker2
sudo systemctl restart docker
2. 启动容器命令(必须用 --runtime=nvidia
bash 复制代码
docker run --runtime=nvidia -it nvidia/cuda:12.0-base nvidia-smi

这样就能在 Docker 18.09 里用 GPU


给你总结最关键的

  • Docker 18.09.7 → 不支持 --gpus
  • 但支持 nvidia-docker2 → 用 --runtime=nvidia 就能调用宿主机 GPU
  • 性能和新版完全一样,只是命令不同

你如果不想升级 Docker

我可以直接给你:

  • 18.09 可用的 docker-compose.yml 配置 GPU
  • 18.09 可用的 FFmpeg + NVENC 硬解容器

需要吗?

好,我直接给你 Docker 18.09.7 能用、完全适配旧版 nvidia-docker2 的配置,包括:

  • docker-compose.yml 写法
  • 启动命令
  • 验证 GPU 是否成功的方法
  • 适合你音视频/WebRTC 场景的 FFmpeg 硬编解示例

全程不用升级 Docker。


1. 先确认你装好了 nvidia-docker2

如果还没装,执行这几句(复制粘贴即可):

bash 复制代码
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list

sudo apt update
sudo apt install -y nvidia-docker2
sudo systemctl restart docker

2. docker-compose.yml 配置(关键)

Docker 18.09 不支持 deploy.resources,只能用 runtime: nvidia

yaml 复制代码
version: '2.3'  # 必须用 2.3,不能用 3.x

services:
  media-server:
    image: 你的镜像
    runtime: nvidia  # 核心:启用GPU
    environment:
      - NVIDIA_VISIBLE_DEVICES=all  # 使用所有GPU
      # 或只使用GPU0
      # - NVIDIA_VISIBLE_DEVICES=0
    ports:
      - "xxxx:xxxx"
    volumes:
      - ./xxx:/xxx

重点:

  • version: '2.3' 必须写
  • runtime: nvidia
  • 用环境变量 NVIDIA_VISIBLE_DEVICES 控制显卡

启动命令还是你原来的:

bash 复制代码
docker-compose up -d

3. 测试容器是否能看到 GPU

bash 复制代码
docker run --rm --runtime=nvidia nvidia/cuda:11.4.1-base nvidia-smi

能正常显示显卡型号、驱动版本 → 成功。


4. FFmpeg 使用 NVENC 硬编、NVDEC 硬解(你音视频场景)

在容器内执行:

硬解码 H.264

bash 复制代码
ffmpeg -hwaccel nvdec -i input.mp4 -f null -

硬编码 H.264

bash 复制代码
ffmpeg -i input.mp4 -c:v h264_nvenc -b:v 5M output.mp4

硬编码 H.265 (HEVC)

bash 复制代码
ffmpeg -i input.mp4 -c:v hevc_nvenc -b:v 5M output.mp4

只要不报错,就是 GPU 正常工作。


5. 如果你用的是 Intel 核显(HD 630)

不用 nvidia,直接直通设备即可,docker-compose.yml:

yaml 复制代码
version: '2.3'
services:
  media-server:
    image: 你的镜像
    devices:
      - /dev/dri:/dev/dri  # 直通核显

容器内用 VA-API 硬解:

bash 复制代码
ffmpeg -hwaccel vaapi -i input.mp4 ...

最简单总结(你记这两句就行)

  • Docker 18.09 → 不能用 --gpus / deploy
  • 替代方案 → runtime: nvidia + NVIDIA_VISIBLE_DEVICES
  • docker-compose 必须用 version: '2.3'

你把你的 docker-compose.yml 发给我,我可以直接帮你改成能跑 GPU 的版本。

相关推荐
cyber_两只龙宝2 小时前
【Oracle】Oracle之SQL的集合运算符
linux·运维·数据库·sql·云原生·oracle
小小的木头人2 小时前
Nginx 访问控制及安全配置文档
运维·nginx·安全
菱玖2 小时前
K8s集群部署与应用运维实战
运维·容器·kubernetes
Elastic 中国社区官方博客2 小时前
自动化可靠性:自愈型企业的架构
运维·elasticsearch·搜索引擎·云原生·架构·自动化·serverless
枕布响丸辣2 小时前
万字详解 GlusterFS 分布式文件系统:原理 + 卷类型 + CentOS 7 集群部署
linux·运维·centos
程序员大辉2 小时前
开源客户端SSH Netcatty:免费替代Termius,带AI的现代化运维工具
运维·开源·ssh
Elivs.Xiang2 小时前
ubuntu20中安装Jenkins
linux·运维·ubuntu·jenkins
说再见再也见不到2 小时前
Ubuntu 将阿里云 OSS 对象存储挂载为本地硬盘(含开机自启)
linux·运维·服务器·ubuntu·阿里云·云计算
坚持就完事了2 小时前
Linux的重定向符
运维·服务器·前端