一机多卡如何指定gpu

Open a1340690674 opened this issue 11 months ago • 1 comments

请教一下，在使用的时候如何指定哪张gpu

Mar 17 '25 06:03 a1340690674

待我有空，debug看一下哈

Mar 19 '25 00:03 SWHL

这个已经在rapid_layout==1.0.0中可以通过参数指定了。示例：

from rapid_layout import EngineType, ModelType, RapidLayout, RapidLayoutInput

cfg = RapidLayoutInput(
    model_type=ModelType.PP_LAYOUT_CDLA,
    engine_type=EngineType.ONNXRUNTIME,
    engine_cfg={"use_cuda": True, "cuda_ep_cfg.gpu_id": 1},  # 这里指定
)
layout_engine = RapidLayout(cfg=cfg)

img_path = "https://raw.githubusercontent.com/RapidAI/RapidLayout/718b60e927ab893c2fad67c98f753b2105a6f421/tests/test_files/layout.jpg"
results = layout_engine(img_path)
print(results)

results.vis("layout_res.png")

Jul 14 '25 00:07 SWHL