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)