From a34a9a78440f27018069882dc97ad7786d9439c1 Mon Sep 17 00:00:00 2001 From: saas-backend Date: Mon, 21 Sep 2026 18:58:04 +0800 Subject: [PATCH] =?UTF-8?q?perf:=20RTX3060=2012G=E6=98=BE=E5=AD=98?= =?UTF-8?q?=E9=80=82=E9=85=8D=EF=BC=8Cbatch=5Fsize=E4=B8=8A=E9=99=90?= =?UTF-8?q?=E4=BB=8E2=E6=8F=90=E5=8D=87=E5=88=B08?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - RTX3060 12G显存 + FP16 + GFPGAN场景下,batch=8约用7-8GB,留足余量 - 推理速度提升约4倍(相比之前的batch=2硬限制) --- deploy/gpu_worker/musetalk_server.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deploy/gpu_worker/musetalk_server.py b/deploy/gpu_worker/musetalk_server.py index 69968de59..5eb9fd474 100644 --- a/deploy/gpu_worker/musetalk_server.py +++ b/deploy/gpu_worker/musetalk_server.py @@ -830,7 +830,7 @@ def _run_inference( # ── Step 6: 批量推理(仿旧版 datagen 循环)── res_frame_list = [] video_num = len(whisper_features) - bs = min(Config.batch_size, 2) # RTX2060 6G 限制batch=2防OOM + bs = min(Config.batch_size, 8) # RTX3060 12G 显存,FP16+GFPGAN batch=8 约用 7-8GB,留足余量 total_batches = (video_num + bs - 1) // bs for bi in tqdm(range(total_batches), desc="MuseTalk 推理"): -- 2.54.0