单条推理转批量推理prompt

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

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

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

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

加入多线程:

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

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

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

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

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

相关推荐
wyg_0311132 分钟前
机器问道:大模型RAG 解读凡人修仙传
人工智能·python·transformer
未来之窗软件服务2 分钟前
幽冥大陆(七十九)Python 水果识别训练视频识别 —东方仙盟练气期
开发语言·人工智能·python·水果识别·仙盟创梦ide·东方仙盟
光影少年20 分钟前
AI前端开发需要会哪些及未来发展?
前端·人工智能·前端框架
hqyjzsb24 分钟前
2026年AI证书选择攻略:当“平台绑定”与“能力通用”冲突,如何破局?
大数据·c语言·人工智能·信息可视化·职场和发展·excel·学习方法
独自归家的兔25 分钟前
基于 cosyvoice-v3-plus 的简单语音合成
人工智能·后端·语音复刻
民乐团扒谱机25 分钟前
【微实验】Python——量子增强时频传递的精度量化
人工智能·python·aigc·量子力学·时空·参数敏感性·光量子
G***技27 分钟前
杰和IB3-771:以RK3588赋能机场巡检机器人
人工智能·物联网
xinyaokeji1 小时前
认准高精度:基恩士 VL 扫描仪为三维测量优选之选
大数据·人工智能
mubei-1231 小时前
万字RAG综述:大语言模型的检索增强生成
人工智能·llm·rag·检索增强生成
Java中文社群1 小时前
国内直连GPT、Claude和Gemini?N8N这次更新真的绝了!
人工智能·后端