- 将 gpu_ids 列表从 [0, 1] 修改为 [0, 1, 2, 3] - 此修改使得程序能够利用更多 GPU 资源, potentially 提高处理能力
@@ -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,取消下面的注释