PaddleHub 如何实现多卡多进程预测?关于`use_gpu: true`写在`config`外层不能正常启动服务,

hiz5n14c  于 6个月前  发布在  其他
关注(0)|答案(1)|浏览(51)

根据您的描述,您希望在PaddleHub 2.3.1和PaddlePaddle 2.4.2.post112环境下使用多卡多进程进行OCR识别。您可以尝试以下配置:

{
 "modules_info": {
  "ocr_system": {
   "init_args": {
    "version": "1.0.0",
    "use_gpu": true
   },
   "predict_args": {},
   "config": {
    "model_list": [
     // 这里添加您的模型列表,例如:
     // {"name": "your_model", "url": "https://paddlehub.io/models/your_model"}
    ],
    "batch_size": 8,
    "use_gpu": true,
    "gpus": "0,1,2,3"
   }
  }
 },
 "port": 8866,
 "use_multiprocess": true,
 "use_gpu": true,
 "gpus": "0,1,2,3"
}

请注意将"model_list"中的模型名称替换为您要使用的模型名称,并确保已正确安装这些模型。同时,您可以根据需要调整"batch_size"和其他参数。

cgvd09ve

cgvd09ve1#

PaddleHub已经停止更新了,更多OCR的预测流程,请参考:https://github.com/PaddlePaddle/PaddleOCR

相关问题