opensource/PaddleOCR
3
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2026-09-24 23:33:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix gpu_id

Browse Source 
fix gpu_id
This commit is contained in:
Thomas Young
2021-05-21 20:29:05 +08:00
committed by GitHub
parent 6893d151b9
commit e0d7deaab3
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
deploy/pdserving/win/ocr_web_server.py
+1 -1
View File
@@ -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(