diff --git a/main.py b/main.py index e90a61a..660a77d 100644 --- a/main.py +++ b/main.py @@ -9,7 +9,7 @@ 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" +os.environ["CUDA_VISIBLE_DEVICES"] = "1" # 项目必须从此处启动,否则代码中的相对路径可能导致错误的发生 if __name__ == '__main__':