From 8e7745f1f6f72ead29856cdc7938718dd151c6cd Mon Sep 17 00:00:00 2001 From: liuyebo <1515783401@qq.com> Date: Thu, 17 Oct 2024 17:08:29 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BD=BF=E7=94=A8gpu=E8=BF=90=E8=A1=8C?= =?UTF-8?q?=E6=96=87=E6=A1=A3=E6=A3=80=E6=B5=8B=E6=A8=A1=E5=9E=8B?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- services/paddle_services/paddle_detection/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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}])