aperimau/nvidia-cuda-ffmpeg

docker-nvidia-cuda-ffmpeg

镜像概述和主要用途

docker-nvidia-cuda-ffmpeg是一个集成了FFmpeg的Docker容器，专门针对NVIDIA GPU进行优化，支持多种CUDA加速的滤镜和编码器。该镜像主要用于需要GPU加速的视频处理任务，如视频转码、实时滤镜应用、多视频拼接（Mosaic）及RTSP摄像头流处理等场景，能够显著提升视频处理效率。

核心功能和特性

• CUDA加速滤镜：支持scale_cuda（GPU加速缩放）、overlay_cuda（GPU加速叠加）等滤镜
• 硬件编码器：集成hevc_nvenc（H.265硬件编码）和h264_nvenc（H.264硬件编码）
• 视频拼接功能：内置Mosaic工具，支持4、6、9路视频源拼接成单画面输出
• 摄像头流处理：提供Camera命令，简化RTSP摄像头流的转码和转发流程
• 灵活配置：通过环境变量自定义输出参数，支持生成自定义FFmpeg命令

使用场景和适用范围

• 实时视频转码服务
• 多摄像头视频监控画面拼接
• RTSP摄像头流转发与处理
• 需要GPU加速的视频处理工作流
• 视频直播流的实时处理

使用方法和配置说明

查看滤镜/编码器帮助

scale_cuda滤镜

bash
docker run -it --rm --gpus=all -e NVIDIA_VISIBLE_DEVICES=all -e NVIDIA_DRIVER_CAPABILITIES=all ghcr.io/aperim/nvidia-cuda-ffmpeg:latest -h filter=scale_cuda

overlay_cuda滤镜

bash
docker run -it --rm --gpus=all -e NVIDIA_VISIBLE_DEVICES=all -e NVIDIA_DRIVER_CAPABILITIES=all ghcr.io/aperim/nvidia-cuda-ffmpeg:latest -h filter=overlay_cuda

hevc_nvenc编码器

bash
docker run -it --rm --gpus=all -e NVIDIA_VISIBLE_DEVICES=all -e NVIDIA_DRIVER_CAPABILITIES=all ghcr.io/aperim/nvidia-cuda-ffmpeg:latest -h encoder=hevc_nvenc

h264_nvenc编码器

bash
docker run -it --rm --gpus=all -e NVIDIA_VISIBLE_DEVICES=all -e NVIDIA_DRIVER_CAPABILITIES=all ghcr.io/aperim/nvidia-cuda-ffmpeg:latest -h encoder=h264_nvenc

Mosaic视频拼接功能

Mosaic功能可将4、6或9路视频源拼接成单画面输出，支持自定义输出参数和目标地址。

环境变量配置

Docker Compose示例

yaml
---
version: '3.8'

services:
  ffmpeg:
    image: ghcr.io/aperim/nvidia-cuda-ffmpeg:latest
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]
    restart: unless-stopped              
    environment:
      PUID: 1000
      PGID: 1000
      TZ: Australia/Sydney
      NVIDIA_VISIBLE_DEVICES: all
      NVIDIA_DRIVER_CAPABILITIES: all
      NVIDIA_REQUIRE_CUDA: cuda>=11.4
      INPUT1: https://cph-p2p-msl.akamaized.net/hls/live/2000341/test/master.m3u8
      INPUT2: https://d3rlna7iyyu8wu.cloudfront.net/skip_armstrong/skip_armstrong_multichannel_subs.m3u8
      INPUT3: http://amssamples.streaming.mediaservices.windows.net/634cd01c-6822-4630-8444-8dd6279f94c6/CaminandesLlamaDrama4K.ism/manifest(format=m3u8-aapl)
      INPUT4: https://devimages.apple.com.edgekey.net/iphone/samples/bipbop/bipbopall.m3u8
    volumes:
      - "/etc/timezone:/etc/timezone:ro"
      - "/etc/localtime:/etc/localtime:ro"
    command: mosaic

生成自定义命令示例

bash
docker run -it --rm \
    -e GENCMD=1 \
    -e INPUT1=https://cph-p2p-msl.akamaized.net/hls/live/2000341/test/master.m3u8 \
    -e INPUT2=https://d3rlna7iyyu8wu.cloudfront.net/skip_armstrong/skip_armstrong_multichannel_subs.m3u8 \
    -e INPUT3=http://amssamples.streaming.mediaservices.windows.net/634cd01c-6822-4630-8444-8dd6279f94c6/CaminandesLlamaDrama4K.ism/manifest\(format=m3u8-aapl\) \
    -e INPUT4=https://devimages.apple.com.edgekey.net/iphone/samples/bipbop/bipbopall.m3u8 \
    ghcr.io/aperim/nvidia-cuda-ffmpeg:latest mosaic

Camera RTSP流处理功能

Camera命令用于简化RTSP摄像头流的转码和转发，支持自定义码率、编码器和帧率。

使用语法

bash
docker run -it --rm --gpus=all -e NVIDIA_VISIBLE_DEVICES=all -e NVIDIA_DRIVER_CAPABILITIES=all ghcr.io/aperim/nvidia-cuda-ffmpeg:latest camera \
	<camera_url> <destination_url> [bitrate] [encoder] [framerate]

参数说明

• camera_url: 摄像头RTSP地址（需包含URL编码的用户名/密码）
• destination_url: 目标RTSP地址（必须使用TCP协议）
• bitrate (可选): 输出码率，默认8M
• encoder (可选): 编码器类型，支持h264或hevc，默认hevc
• framerate (可选): 输出帧率

示例

bash
docker run -it --rm --gpus=all -e NVIDIA_VISIBLE_DEVICES=all -e NVIDIA_DRIVER_CAPABILITIES=all ghcr.io/aperim/nvidia-cuda-ffmpeg:latest camera \
	rtsp://user:pass@192.0.2.1/unicast/c1/s0/live rtsp://host.docker.internal:8554/camera1 16M h264 24