网易首页 > 网易号 > 正文 申请入驻

用PyNvVideoCodec和vLLM扩展多GPU视频描述任务

0
分享至

视频描述是很多行业的常见任务,用来描述视频里正在发生什么。自动驾驶训练系统用它来描述和分类危险场景,也能生成人类可读、可搜索的元数据。

此前vLLM处理这类数据集,视频只能通过基于CPU的OpenCV+FFMPEG后端解码。在多GPU节点上(每个GPU跑一个vLLM服务器),CPU要在VLM推理开始前完成视频帧解码,压力陡增。视频描述任务的输出相对较短(100-200个token),解码视频占用的时间比例因此更大,CPU解码很快成为瓶颈,即使只有2或4块GPU,也能把CPU核心跑满。


通过集成PyNvVideoCodec——NVIDIA硬件视频解码器(也叫NVDEC)的Python接口——vLLM把视频解码负载从CPU转移出去,移除了上述瓶颈,即便扩展到8块GPU也能保持良好伸缩性。

怎么在vLLM里用上硬件解码

标准CUDA版vLLM发布包里已经包含了PyNvVideoCodec的功能。如果你用的是自定义安装的vLLM,确保项目里加上对PyNvVideoCodec==2.0.4的PyPi依赖。

启动vLLM时,有几个点需要注意:

  • CUDA MPS对多进程高并发场景(比如批量VLM推理)的性能至关重要,建议在vllm serve之前确认MPS守护进程已启动。
  • --mm-ipc-gpu-memory-gb用于为视频解码预留显存。可以在不同取值下测试吞吐量,只预留不影响吞吐量的最小值。
  • 扩展到多GPU时,通常建议每个vLLM服务器副本跑一个容器,每个容器只暴露一块GPU;也可以用CUDA_VISIBLE_DEVICES给每个vLLM副本暴露单块GPU。请求分发用反向代理完成。

关于PyNvVideoCodec解码器后端的参数文档,可以参考vLLM的相关文档。

多GPU视频描述任务的伸缩表现

NVIDIA自动驾驶团队用视频描述任务做了测试。这些系统要处理数十万小时的视频片段,对应数亿次视频描述请求。

这类任务通常用相对轻量的模型,比如Qwen/Qwen3-VL-8B-Instruct,输入提示词指定想要的描述类型,输出在100-200个token量级。

在8块H100、8个vLLM副本各配一块GPU的配置下,基于GPU的视频解码吞吐量是CPU解码的两倍以上。

另一个值得注意的变化是:此前使用最多8块GPU的大规模工作负载,在不到4块GPU时就会先撞上CPU利用率瓶颈。现在有了硬件视频解码支持,这个CPU瓶颈被移除了。

需要说明的是,视频解码确实需要预留一部分显存。如果你的用例已经把全部显存用于KV缓存,可能会看到一些影响。但在实际测试中,还没有遇到使用PyNvVideoCodec带来性能下降的情况。

一个输入提示词的例子是这样的:分析这段前向行车记录仪视频,简要描述驾驶环境、相关道路使用者和交通控制,以及自车的动作,只报告清晰可见的细节,避免推测。

对应的输出则是:自车在白天沿多车道城市道路行驶,能见度清晰。前方同车道和相邻车道有多辆车,可见一个信号灯路口。自车保持车道,接近车流时减速,并与前车保持距离继续行驶。

特别声明:以上内容(如有图片或视频亦包括在内)为自媒体平台“网易号”用户上传并发布,本平台仅提供信息存储服务。

Notice: The content above (including the pictures and videos if any) is uploaded and posted by a user of NetEase Hao, which is a social media platform and only provides information storage services.

相关推荐
热点推荐
娃哈哈被查封,震惊行业

娃哈哈被查封,震惊行业

李东阳朋友圈
2026-09-18 11:19:09
一家已经消失的手机品牌,做出了史上最好的安卓机

一家已经消失的手机品牌,做出了史上最好的安卓机

灰度测试中
2026-09-17 00:10:07
脸都不要了!中国男篮惨败日本队,三人要淘汰,郭士强必须下课

脸都不要了!中国男篮惨败日本队,三人要淘汰,郭士强必须下课

宗介说体育
2026-09-18 20:20:54
中央政法委发布公示

中央政法委发布公示

政知新媒体
2026-09-18 14:44:25
乌军大规模抗议爆发!泽连斯基承认:俄军随时可以对自己进行斩首

乌军大规模抗议爆发!泽连斯基承认:俄军随时可以对自己进行斩首

观锐器
2026-09-18 12:49:44
美国返还中国58件文物!其中一菩萨像系日本人金条买通住持盗凿,盗贼还低价买空恭王府

美国返还中国58件文物!其中一菩萨像系日本人金条买通住持盗凿,盗贼还低价买空恭王府

不掉线电波
2026-09-18 11:58:04
三名机长吃印度酒店早餐,飞行途中集体拉稀,飞机差点变无人驾驶

三名机长吃印度酒店早餐,飞行途中集体拉稀,飞机差点变无人驾驶

杭城村叔
2026-09-18 13:55:07
宇树科技股价已稳稳跌破500,评论区炸锅…

宇树科技股价已稳稳跌破500,评论区炸锅…

慧翔百科
2026-09-18 11:46:33
敬一丹病逝,丈夫手握3600亿资产、动用顶尖医疗团队,三个月贴身守护仍没能留住她

敬一丹病逝,丈夫手握3600亿资产、动用顶尖医疗团队,三个月贴身守护仍没能留住她

子芫伴你成长
2026-09-17 21:38:13
106比40狂胜66分!女篮亚运再现一边倒屠杀:中国队两大劲敌杀红眼

106比40狂胜66分!女篮亚运再现一边倒屠杀:中国队两大劲敌杀红眼

篮球快餐车
2026-09-18 11:57:01
冲上热搜!多所高校发文“过紧日子”,北大明确禁去21个景区开会,能线上开的会不线下聚

冲上热搜!多所高校发文“过紧日子”,北大明确禁去21个景区开会,能线上开的会不线下聚

南方都市报
2026-09-18 07:58:40
没有对比就没有差距,廖三宁和胡明轩、巅峰赵睿的后卫级别,根本不是一个档次

没有对比就没有差距,廖三宁和胡明轩、巅峰赵睿的后卫级别,根本不是一个档次

林子说事
2026-09-18 08:59:23
随着中国男篮78-97日本 产生3个意外和2个事实 郭士强+王俊杰上榜

随着中国男篮78-97日本 产生3个意外和2个事实 郭士强+王俊杰上榜

侃球熊弟
2026-09-18 20:05:27
广东佛山一男子高空作业从30楼下到2楼,发现绳子被2楼老人剪断,当事人:老人不知是高空作业,想当晾衣绳用,已赔偿费用

广东佛山一男子高空作业从30楼下到2楼,发现绳子被2楼老人剪断,当事人:老人不知是高空作业,想当晾衣绳用,已赔偿费用

潇湘晨报
2026-09-18 15:47:14
崩盘!中国男篮半场落后日本19分 被轰11-0全队无人得分上双

崩盘!中国男篮半场落后日本19分 被轰11-0全队无人得分上双

醉卧浮生
2026-09-18 19:01:58
太尴尬了!上海豫园的汉服女生对国人全程无视,老外开口就答应,一段路人求合影视频引争议

太尴尬了!上海豫园的汉服女生对国人全程无视,老外开口就答应,一段路人求合影视频引争议

火山詩话
2026-09-18 05:55:55
私吞上亿善款终于尘埃落定?官方正式宣布,54岁韩红身份迎来转变

私吞上亿善款终于尘埃落定?官方正式宣布,54岁韩红身份迎来转变

阿伧说事
2026-09-15 15:58:58
委内瑞拉即将达成协议 将40亿美元黄金储备从英国转移至美国

委内瑞拉即将达成协议 将40亿美元黄金储备从英国转移至美国

财联社
2026-09-18 12:47:03
中国男篮惨败!媒体人热议:比国足输越南更离谱,郭士强下课吧

中国男篮惨败!媒体人热议:比国足输越南更离谱,郭士强下课吧

奥拜尔
2026-09-18 20:13:26
泽连斯基已经向全世界亮明态度:乌克兰可以向俄罗斯投降,也可以放弃加入任何国际组织,但是唯独顿巴斯是坚决不会放弃!

泽连斯基已经向全世界亮明态度:乌克兰可以向俄罗斯投降,也可以放弃加入任何国际组织,但是唯独顿巴斯是坚决不会放弃!

扶苏聊历史
2026-09-18 18:02:45
2026-09-18 21:36:49
我是一个养虾人
我是一个养虾人
有态度网友ytd
295文章数 81关注度
往期回顾 全部

科技要闻

直击iPhone 18新机发售:黄牛加价不如前代

头条要闻

外媒提问"中方领导人是否将访问美国" 外交部正面回应

头条要闻

外媒提问"中方领导人是否将访问美国" 外交部正面回应

体育要闻

一个角色球员,靠什么在NBA征战17年

娱乐要闻

陈思诚 佟丽娅不想让儿子知道两人离婚

财经要闻

研发0.25亿理财26亿,敷尔佳豪赌三类器械

汽车要闻

纯电/插混双动力+五座/六座双布局 海狮08应该怎么选?

态度原创

数码
艺术
本地
亲子
公开课

数码要闻

蓝宝石公布类魂游戏《不朽遗志》联名款NITRO+ RX 9070 XT OC显卡

艺术要闻

文徵明5毫米小楷现身美国,写了三年才完成,这字用放大镜才能看清!

本地新闻

中秋逛白塔寺,体验国医妙荟雅集

亲子要闻

走进爱他美放心工厂:以食品安全守护万千妈妈信任

公开课

李玫瑾:为什么性格比能力更重要?

无障碍浏览 进入关怀版