单条推理转批量推理prompt

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

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

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

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

加入多线程:

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

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

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

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

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

相关推荐
THMAIL1 小时前
深度学习从入门到精通 - 生成对抗网络(GAN)实战:创造逼真图像的魔法艺术
人工智能·python·深度学习·神经网络·机器学习·生成对抗网络·cnn
却道天凉_好个秋1 小时前
计算机视觉(八):开运算和闭运算
人工智能·计算机视觉·开运算与闭运算
无风听海1 小时前
神经网络之深入理解偏置
人工智能·神经网络·机器学习·偏置
JoinApper1 小时前
目标检测系列-Yolov5下载及运行
人工智能·yolo·目标检测
北京地铁1号线1 小时前
GPT(Generative Pre-trained Transformer)模型架构与损失函数介绍
gpt·深度学习·transformer
飞哥数智坊2 小时前
即梦4.0实测:我真想对PS说“拜拜”了!
人工智能
fantasy_arch2 小时前
9.3深度循环神经网络
人工智能·rnn·深度学习
Ai工具分享2 小时前
视频画质差怎么办?AI优化视频清晰度技术原理与实战应用
人工智能·音视频
新智元2 小时前
不到 10 天,国产「香蕉」突袭!一次 7 图逼真还原,合成大法惊呆歪果仁
人工智能·openai
LaughingZhu2 小时前
Product Hunt 每日热榜 | 2025-09-07
人工智能·经验分享·搜索引擎·产品运营