调整显卡调用

docker-compose.yml

@@ -108,7 +108,7 @@ services:
       resources:
         reservations:
           devices:
-            - device_ids: [ "0", "1" ]
+            - device_ids: [ "1" ]
               capabilities: [ "gpu" ]
               driver: "nvidia"
 
@@ -126,7 +126,7 @@ services:
       resources:
         reservations:
           devices:
-            - device_ids: [ "0", "1" ]
+            - device_ids: [ "1" ]
               capabilities: [ "gpu" ]
               driver: "nvidia"
 
@@ -144,6 +144,6 @@ services:
       resources:
         reservations:
           devices:
-            - device_ids: [ "0", "1" ]
+            - device_ids: [ "1" ]
               capabilities: [ "gpu" ]
               driver: "nvidia"

photo_mask/__init__.py

@@ -39,7 +39,7 @@ SIMILAR_CHAR = {
 
 # 如果不希望识别出空格，可以设置use_space_char=False。做此项设置一定要测试，2.7.3版本此项设置有bug，会导致识别失败
 OCR = PaddleOCR(
-    gpu_id=1,
+    gpu_id=0,
     show_log=False,
     det_db_thresh=0.1,
     det_db_box_thresh=0.3,