From 8cc9e79cc319b974ba9631f5d669a584eb8969dd Mon Sep 17 00:00:00 2001
From: liuyebo <1515783401@qq.com>
Date: Tue, 11 Jun 2024 13:41:50 +0800
Subject: [PATCH] =?UTF-8?q?=E5=8F=8C=E5=8D=A1=E9=85=8D=E7=BD=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 main.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/main.py b/main.py
index fa272f3..d764e28 100644
--- a/main.py
+++ b/main.py
@@ -1,4 +1,5 @@
 import logging.config
+import os
 import traceback
 
 from auto_email.error_email import send_an_error_email
@@ -6,6 +7,8 @@ from config.log import LOGGING_CONFIG
 from config.photo_review import RETRY_TIME, SEND_ERROR_EMAIL
 from photo_review.photo_review import main
 
+os.environ["CUDA_VISIBLE_DEVICES"] = "0,1"
+
 # 项目必须从此处启动，否则代码中的相对路径可能导致错误的发生
 if __name__ == '__main__':
     logging.config.dictConfig(LOGGING_CONFIG)