forgetest5
/
PaddleOCR
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix gpu_id 

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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
deploy/pdserving/win/ocr_web_server.py
View File

@ -42,7 +42,7 @@ class OCRService(WebService):
self.det_client = LocalPredictor() self.det_client = LocalPredictor()
if sys.argv[1] == 'gpu': if sys.argv[1] == 'gpu':
self.det_client.load_model_config( 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': elif sys.argv[1] == 'cpu':
self.det_client.load_model_config(det_model_config) self.det_client.load_model_config(det_model_config)
self.ocr_reader = OCRReader( self.ocr_reader = OCRReader(