From 9584d1fe35752f2c78aad3172698acc8d1bd02eb Mon Sep 17 00:00:00 2001 From: liuyebo <1515783401@qq.com> Date: Fri, 21 Jun 2024 14:20:58 +0800 Subject: [PATCH] =?UTF-8?q?=E8=B0=83=E6=95=B4GPU=E9=85=8D=E7=BD=AE?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- config/photo_review.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/config/photo_review.py b/config/photo_review.py index c1d2318..bd0e716 100644 --- a/config/photo_review.py +++ b/config/photo_review.py @@ -17,4 +17,4 @@ IE_BATCH_SIZE = 4 LAYOUT_ANALYSIS = False # 可使用的GPU -CUDA_VISIBLE_DEVICES = "1" +CUDA_VISIBLE_DEVICES = "0,1"