Browse Source

workers_per_gpu = 1 多线程还是内存冲突

zhch158_admin 3 months ago
parent
commit
17497735f3
1 changed files with 1 additions and 1 deletions
  1. 1 1
      zhch/ppstructurev3_dual_gpu_optimized.py

+ 1 - 1
zhch/ppstructurev3_dual_gpu_optimized.py

@@ -322,7 +322,7 @@ def main():
     
     # 双GPU处理参数
     batch_size = 4              # 批处理大小
-    workers_per_gpu = 2         # 每个GPU的worker数量(24GB GPU推荐2个)
+    workers_per_gpu = 1         # 每个GPU的worker数量(24GB GPU推荐2个)
     
     # 确保输出目录存在
     print(f"输出目录: {Path(output_dir).absolute()}")