diff --git a/services/paddle_services/paddle_detection/__init__.py b/services/paddle_services/paddle_detection/__init__.py index b71c46c..9b59f35 100644 --- 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}])