单条推理转批量推理prompt

为了将单条推理程序改为批量推理程序,并实现您的要求,我们需要进行以下步骤:

输入的图片和视频都是随机从视频文件夹、图片文件夹挑选,组成输入对:

需要编写一个函数来读取指定文件夹中的所有图片和视频文件。

使用随机选择机制来组合这些文件作为输入对。

加入多线程:

使用 concurrent.futures.ThreadPoolExecutor 来管理多线程任务。

让每个GPU都固定加载需要的模型,避免重复加载、释放模型:

在初始化时为每个GPU加载一次模型,并在多线程中复用这些模型。

生成线程时动态选择GPU,如果选择好了,这个线程全周期一直在当前GPU,防止张量、数据不在同一块GPU:

在每个线程中设置环境变量 CUDA_VISIBLE_DEVICES,以确保每个线程只使用指定的GPU。

相关推荐
美狐美颜sdk9 小时前
直播APP开发技术架构解析:从音视频流到第三方美颜SDK接入的完整方案
大数据·人工智能·音视频·美颜sdk·美颜api
CIO_Alliance9 小时前
AI认知系列(3)| 数据、算法、算力、场景四要素协同
人工智能·算法·ipaas·系统集成·企业cio联盟·企业级ai化转型
技术传感器9 小时前
让 Hermes 自动完成一个软件需求:从 Issue 到 PR 的 AI 开发流水线
人工智能·架构·aigc·需求分析·issue
啥都想学点的研究生9 小时前
一篇文章讲清楚:机器学习的相关概念
人工智能·机器学习
Capricorn19889 小时前
知芽研究问题看板无法推进?从孵化区到笔记的机制排查指南
大数据·论文阅读·人工智能·笔记·学习方法·论文笔记
Wang's Blog9 小时前
AI Agent白手起家76:使用 CrewAI 搭建多智能体营销策略生成器
人工智能
梦想的旅途29 小时前
企微 API 二次开发:利用 AI Agent 实现自动化运维
运维·人工智能·企业微信
DS随心转APP9 小时前
deepseek生成的word怎么下载 AI导出鸭全平台方案技术深度测评
人工智能·ai·chatgpt·word·deepseek·ai导出鸭
睿本云10 小时前
从“有数据”到“会用数据”,一次标签体系的升级能带来的改变
大数据·人工智能
俊哥V10 小时前
每日 AI 研究简报 · 2026-08-14
人工智能·ai