From e0d7deaab38bffa95491effd32c84a0e4a003692 Mon Sep 17 00:00:00 2001
From: Thomas Young <35565423+HexToString@users.noreply.github.com>
Date: Fri, 21 May 2021 20:29:05 +0800
Subject: [PATCH] fix gpu_id

fix gpu_id
---
 deploy/pdserving/win/ocr_web_server.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/deploy/pdserving/win/ocr_web_server.py b/deploy/pdserving/win/ocr_web_server.py
index 1f1394fb..9fc94901 100644
--- a/deploy/pdserving/win/ocr_web_server.py
+++ b/deploy/pdserving/win/ocr_web_server.py
@@ -42,7 +42,7 @@ class OCRService(WebService):
         self.det_client = LocalPredictor()
         if sys.argv[1] == 'gpu':
             self.det_client.load_model_config(
-                det_model_config, use_gpu=True, gpu_id=1)
+                det_model_config, use_gpu=True, gpu_id=0)
         elif sys.argv[1] == 'cpu':
             self.det_client.load_model_config(det_model_config)
         self.ocr_reader = OCRReader(
-- 
GitLab