croncorp/python-ffmpeg

python-ffmpeg Docker镜像

镜像概述

python-ffmpeg是一个集成FFmpeg的Docker镜像，基于官方Python 3.9.5-slim-buster镜像构建，专为需要CUDA加速的视频处理任务设计。镜像选择Debian系统作为基础，以确保与特定环境下的CUDA库和API兼容性，适用于需要GPU加速的视频转码、处理场景。

构建的Docker镜像可在 https://hub.docker.com/r/croncorp/python-ffmpeg 获取。

核心功能与特性

• 基础镜像：基于官方python:3.9.5-slim-buster，无需手动在Debian中安装该Python版本
• 系统兼容性：采用Debian系统，通过backports安装软件包，确保与CUDA库和API的兼容性
• CUDA支持：集成CUDA相关依赖，支持GPU加速的视频编解码操作
• FFmpeg集成：包含FFmpeg工具，支持通过CUDA硬件加速进行视频处理

使用前提

1. 宿主环境要求：

   • 宿主系统需为Debian 10，并已更新至backports
   • 需安装backports中的NVIDIA驱动

2. Docker环境配置：

   • 需按照 https://github.com/NVIDIA/nvidia-container-runtime 配置Docker环境

3. CUDA库安装：

   • 宿主需安装以下CUDA库（来自buster-backports）：
     • libcuda1
     • libcudart11.0

注意：需确保宿主与容器内的CUDA API版本一致，避免因版本差异导致冲突。容器会获取仓库中最新的库版本，若宿主未同步更新，可能导致CUDA功能异常。

使用方法

启动容器

通过以下命令启动容器，映射视频目录并启用GPU支持：

bash
docker run -ti -v /video:/mnt --gpus all wladimi/python-ffmpeg:3.9.5-slim-buster bash

参数说明：

• -v /video:/mnt：将宿主的/video目录映射到容器内的/mnt目录
• --gpus all：启用所有GPU设备

容器内FFmpeg使用示例

以下示例展示如何使用CUDA加速将视频转码为480p分辨率：

bash
ffmpeg -y -hwaccel cuvid -c:v h264_cuvid -i "/mnt/Окружающий_мир__3_класс_55урок.mp4" -vf scale_npp="-2:480" -c:v h264_nvenc -preset slow "/tmp/SampleVideo_1280x720_10mbk9g38pcr_480p.mp4"

命令说明：

• -hwaccel cuvid：使用CUDA加速解码
• -c:v h264_cuvid：使用h264_cuvid解码器
• -vf scale_npp="-2:480"：使用NVIDIA Performance Primitives进行视频缩放，保持宽高比，高度为480px
• -c:v h264_nvenc：使用NVIDIA编码器
• -preset slow：设置编码速度/质量平衡为"slow"（质量更高，速度较慢）