请问如何指定使用GPU呢？是否有类似gpu_num的参数？

[wwfra]

1.我在运行的时候感觉gpu没多大显存占用，如何知道是否使用的是gpu？
2.是否有类似ocr_engine = PaddleOCR(use_gpu=True)这样的参数？
3.是否可以指定gpu_num？

[wufan-tb]

1. 'nvidia-smi'可以查看gpu上正在运行的进程，可以看一下；（如果感觉跑的非常非常慢，大概率是没有用到gpu）；
2. ocr任务当前支持的就是paddle的模型，可以在配置文件中传paddle的参数；其他任务应该默认会使用gpu，gpu不可用才会尝试cpu；
3. 暂时还不支持多卡来推理

[wwfra]

paddleocr可以指定gpu，主要是版面检测这部分，有办法指定吗？使用的是默认的docyolo那个模型，似乎是cpu在跑

[JulioZhao97]

@wwfra 您好这个问题现在已经修复，请麻烦pull最新代码，doclayout-yolo的设备指定可以在config里面指定例如device:0