From 8a56308b03241dfd04f7b508cfc9915bc15363a8 Mon Sep 17 00:00:00 2001
From: liuyebo <1515783401@qq.com>
Date: Fri, 2 Aug 2024 09:54:06 +0800
Subject: [PATCH] =?UTF-8?q?=E8=B0=83=E6=95=B4=E6=98=BE=E5=8D=A1=E8=B0=83?=
 =?UTF-8?q?=E7=94=A8?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 docker-compose.yml     | 6 +++---
 photo_mask/__init__.py | 2 +-
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/docker-compose.yml b/docker-compose.yml
index 96aa3d0..7a21534 100644
--- a/docker-compose.yml
+++ b/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"
\ No newline at end of file
diff --git a/photo_mask/__init__.py b/photo_mask/__init__.py
index df55e17..c019dec 100644
--- a/photo_mask/__init__.py
+++ b/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,