当前位置: 首页 > article >正文

单条推理转批量推理prompt

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

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

需要编写一个函数来读取指定文件夹中的所有图片和视频文件。
使用随机选择机制来组合这些文件作为输入对。
加入多线程:

使用 concurrent.futures.ThreadPoolExecutor 来管理多线程任务。
让每个GPU都固定加载需要的模型,避免重复加载、释放模型:

在初始化时为每个GPU加载一次模型,并在多线程中复用这些模型。
生成线程时动态选择GPU,如果选择好了,这个线程全周期一直在当前GPU,防止张量、数据不在同一块GPU:

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


http://www.kler.cn/a/402776.html

相关文章:

  • Linux的开发工具(三)
  • 失落的Apache JDBM(Java Database Management)
  • DataGear 企业版 1.3.0 发布,数据可视化分析平台
  • 社团管理智能化:SpringBoot技术
  • C语言中const char *字符进行切割实现
  • 鸿蒙操作系统(HarmonyOS)开发的初学者了解和入门
  • AI应用中基于okhttp3实现SSE技术的各种解决方案实现
  • MySQL排序与分页
  • kali打开复制粘贴功能
  • python serializer, model drf通过序列化器, 模型获取mysql 一张表某个字段数据库现存的最大值
  • UE5 第一人称射击项目学习(三)
  • SSM全家桶 1.Maven
  • 联邦学习安全聚合算法综述(论文解析)以及如何确定自己研究方向的方法
  • 高频面试-cookie, token, session
  • Feed流系统重构:架构篇
  • 力扣整理版八:回溯算法(待更新)
  • 3C产品说明书电子化转变:用户体验、环保与商业机遇的共赢
  • 使用 Oracle.DataAccess.Client 驱动 和 OleDB 调用Oracle 函数的区别
  • 【JAVA】Java基础—面向对象编程:常用API与数据结构—集合框架(List、Set、Map等)
  • 【Leecode】Leecode刷题之路第56天之合并区间
  • git创建远程仓库,以gitee码云为例GitHub同理
  • Go 1.23 is released(翻译)
  • c语言数据结构与算法--简单实现线性表(顺序表+链表)的插入与删除
  • pcap_set_timeout()函数
  • 文件 IO:高效读取文件的一些注意事项
  • SQL面试题——复杂窗口定义