fixed a cuda config bug

2023-01-27 13:41:39 +08:00 · 2023-01-27 13:41:39 +08:00 · 1b4dfa4449
commit 1b4dfa4449
parent 6fed221f74
1 changed files with 1 additions and 1 deletions
--- a/main.py
+++ b/main.py
@ -23,7 +23,7 @@ class number(object):
        self.debug = debug

        # Init detector and OCR
-        self.__detector = Detector(device="gpu" if gpu else "cpu")
+        self.__detector = Detector(device="cuda" if gpu else "cpu")
        self.__eocr = easyocr.Reader(['ch_sim', 'en'], gpu=self.gpu)
        self.__pocr = PaddleOCR(use_angle_cls=True, use_gpu=self.gpu, show_log=debug)