单条推理转批量推理prompt

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

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

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

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

加入多线程:

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

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

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

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

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

相关推荐
2501_93367079几秒前
2027风控策略岗秋招准备:SQL、Excel、建模的优先级与项目路径
人工智能·sql·excel
小宋10219 分钟前
AI 生成内容如何证明来源:C2PA、Content Credentials 与签名校验
人工智能
zhousenshan10 分钟前
AgentScope 和 Spring AI Alibaba的区别
人工智能
奈落2411 分钟前
【RAG 深度修炼】专栏 · 第 9 期(收官):安全专题与生产 Checklist 总集——从能跑的 Demo 到睡得着觉的生产系统
大数据·网络·人工智能·安全·ai编程
超大青花鱼11 分钟前
Ubuntu20.04安装CUDA11.8教程
人工智能·深度学习·计算机视觉
golang学习记19 分钟前
Jetbrains 正式官宣:全新AI IDE正式发布!
ide·人工智能·intellij-idea·air
HZjiangzi20 分钟前
深耕精密驱动二十六载,捷昌驱动助力机器人产业升级发展
人工智能·算法
鬓戈30 分钟前
Claude Code frontend-design 插件调研 与 Vue 旧系统风格一致性方案
前端·vue.js·人工智能
BSD_CGQ34 分钟前
柔性触觉到工业称重:FSR 高精度工程化适配路径解析
大数据·人工智能·压力传感器·柔性薄膜压力传感器
一隅论数智35 分钟前
给AI找对“富矿“:本体协同的六大应用模式与落地战法
大数据·人工智能·经验分享·笔记·学习·学习方法·政务