调整显存占用

docker-compose.yml

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

photo_mask/__init__.py

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