使用gpu运行文档检测模型

2024-10-17 17:08:29 +08:00
parent cc53243647
commit 8e7745f1f6
1 changed files with 1 additions and 1 deletions
--- a/services/paddle_services/paddle_detection/init.py
+++ b/services/paddle_services/paddle_detection/init.py
@@ -4,4 +4,4 @@ from onnxruntime import InferenceSession

 MODEL_DIR = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), 'model', 'object_det_model')
 PADDLE_DET = InferenceSession(os.path.join(MODEL_DIR, 'ppyoloe_plus_crn_l_80e_coco_w_nms.onnx'),
-                              providers=['CPUExecutionProvider'], provider_options=[{'device_id': 0}])
+                              providers=['CUDAExecutionProvider'], provider_options=[{'device_id': 0}])