如何在GStreamer中使用decodebin指定GPU设备？

可以通过decodebin结合GStreamer的特性实现指定GPU设备的需求，以下是两种实用方案：

1. 通过Caps过滤强制绑定带设备编号的解码器 #

decodebin会根据输入流的编码格式自动匹配解码器，你可以通过设置decodebin的caps参数，限定只使用带有目标GPU编号后缀的解码器（比如nvh264device1dec对应第二块GPU）。

示例命令行管道：

gst-launch-1.0 decodebin uri=file:///path/to/your/video.mp4 caps="video/x-h264, stream-format=(string)avc" ! queue ! nvh264device1dec ! autovideosink

通过caps过滤出H.264流，让decodebin只能匹配到你指定的带设备编号的解码器，从而固定使用第二块GPU。

2. 动态设置解码器的device-id属性 #

连接decodebin的pad-added信号，在解码器实例化完成后，直接为其设置device-id属性指定GPU编号（编号从0开始，1对应第二块GPU）。

以Python代码为例：

import gi
gi.require_version('Gst', '1.0')
from gi.repository import Gst

Gst.init(None)

def on_pad_added(decodebin, pad):
    caps_str = pad.get_current_caps().to_string()
    if 'video/x-h264' in caps_str or 'video/x-h265' in caps_str:
        decoder = pad.get_parent_element()
        # 检查解码器是否支持device-id属性
        if decoder.get_property('device-id') is not None:
            decoder.set_property('device-id', 1)
        # 后续连接视频输出元素
        sink_pad = queue.get_static_pad('sink')
        pad.link(sink_pad)

pipeline = Gst.Pipeline.new("gpu-decode-pipeline")
decodebin = Gst.ElementFactory.make("decodebin", "decode-bin")
decodebin.set_property("uri", "file:///path/to/your/video.mp4")
queue = Gst.ElementFactory.make("queue", "queue")
autovideosink = Gst.ElementFactory.make("autovideosink", "video-sink")

pipeline.add(decodebin, queue, autovideosink)
queue.link(autovideosink)
decodebin.connect("pad-added", on_pad_added)

pipeline.set_state(Gst.State.PLAYING)
bus = pipeline.get_bus()
bus.timed_pop_filtered(Gst.CLOCK_TIME_NONE, Gst.MessageType.ERROR | Gst.MessageType.EOS)
pipeline.set_state(Gst.State.NULL)

这种方式更灵活，不需要硬编码解码器名称，适配不同编码格式的视频流。

补充：全局环境变量绑定 #

如果只需要单管道绑定特定GPU，可以用NVIDIA的CUDA_VISIBLE_DEVICES环境变量全局限制可用GPU，比如：

CUDA_VISIBLE_DEVICES=1 gst-launch-1.0 decodebin uri=file:///path/to/your/video.mp4 ! autovideosink

该命令会让整个GStreamer进程只能看到第二块GPU，decodebin自动选择的解码器也会运行在这块GPU上。

内容的提问来源于stack exchange，提问作者Scrimpy