docker 容器中 ffmpeg 调用 GPU(nvidia)

目标：在 docke 容器中使用 ffmpeg 并可以调用 GPU(nvidia)

构建镜像

• DockerfileFROM nvidia/cuda:11.1.1-base-ubuntu20.04 ENV TZ=Asia/Shanghai \ LANG=C.UTF-8 RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone \ && sed -i 's/archive.ubuntu.com/mirrors.aliyun.com/g' /etc/apt/sources.list \ && sed -i 's/security.ubuntu.com/mirrors.aliyun.com/g' /etc/apt/sources.list \ && apt-get update && apt install -y apt-utils RUN DEBIAN_FRONTEND=noninteractive apt-get install -y \ ca-certificates \ python3 \ python3-pip \ nvidia-driver-470 \ ffmpeg=7:4.2.4-1ubuntu0.1 \ && rm -rf /var/lib/apt/lists/*
• 构建

docker build -t ffmpeg:4.2.4 .

镜像使用

• 启动容器（也有使用 alias 方式）

docker run --rm -it -v $(pwd):/data --gpus all --user $UID -w /data ffmpeg:4.2.4 bash

• 在容器中使用 ffmpeg

ffmpeg -i input
# 使用 cuda
ffmpeg -hwaccel cuda -i input