fix cuda default value (#1051)

0cda9b87 · littletomatodonkey · GitHub · 8477aa8e · 0cda9b87
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

ppocr/data/cls/dataset_traversal.py ppocr/data/cls/dataset_traversal.py +1 -1

未找到文件。
--- a/ppocr/data/cls/dataset_traversal.py
+++ b/ppocr/data/cls/dataset_traversal.py
@@ -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: