浏览代码

feat(zhch): 增加可用 GPU 数量

- 将 gpu_ids 列表从 [0, 1] 修改为 [0, 1, 2, 3]
- 此修改使得程序能够利用更多 GPU 资源, potentially 提高处理能力
zhch158_admin 3 月之前
父节点
当前提交
09d99f9a8a
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      zhch/ppstructurev3_multi_gpu_multiprocess.py

+ 1 - 1
zhch/ppstructurev3_multi_gpu_multiprocess.py

@@ -351,7 +351,7 @@ def main():
     
     # 多GPU多进程参数(可配置)
     batch_size = 4                  # 批处理大小
-    gpu_ids = [0, 1]               # 指定使用的GPU ID列表 - 可修改
+    gpu_ids = [0, 1, 2, 3]               # 指定使用的GPU ID列表 - 可修改
     pipelines_per_gpu = 1          # 每个GPU的pipeline实例数 - 可修改
     
     # 如果想要自动检测所有可用GPU,取消下面的注释