Update dataset_traversal.py

CUDA_VISIBLE_DEVICES 默认值必须为str, 否则split会报错
5 years ago · b2e1a84981
parent 5ed1e2427b
commit b2e1a84981
1 changed files with 1 additions and 1 deletions
--- a/ppocr/data/rec/dataset_traversal.py
+++ b/ppocr/data/rec/dataset_traversal.py
@ -237,7 +237,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: