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

fix cuda default value (#1051)

This commit is contained in:
littletomatodonkey 2020-10-29 12:57:50 +08:00 committed by GitHub
parent 8477aa8ec3
commit 0cda9b8757
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
ppocr/data/cls/dataset_traversal.py
View File

@ -74,7 +74,7 @@ class SimpleReader(object):
def get_device_num():
if self.use_gpu:
gpus = os.environ.get("CUDA_VISIBLE_DEVICES", 1)
gpus = os.environ.get("CUDA_VISIBLE_DEVICES", "1")
gpu_num = len(gpus.split(','))
return gpu_num
else: