fix gpu_id

fix gpu_id

fix gpu_id
e0d7deaa · Thomas Young · GitHub · 6893d151 · e0d7deaa
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

deploy/pdserving/win/ocr_web_server.py deploy/pdserving/win/ocr_web_server.py +1 -1

未找到文件。
--- 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(