调整指定GPU

This commit is contained in:
2024-06-19 17:11:31 +08:00
parent 501d62f9ce
commit 3b6f965eae
2 changed files with 6 additions and 10 deletions

10
main.py
View File

@@ -2,15 +2,15 @@ import logging.config
import os
import traceback
import paddle
from auto_email.error_email import send_an_error_email
from config.log import LOGGING_CONFIG
from config.photo_review import RETRY_TIME, SEND_ERROR_EMAIL, CUDA_VISIBLE_DEVICES
from photo_review.photo_review import main
os.environ["CUDA_VISIBLE_DEVICES"] = CUDA_VISIBLE_DEVICES
import paddle
from auto_email.error_email import send_an_error_email
from config.log import LOGGING_CONFIG
from photo_review.photo_review import main
# 项目必须从此处启动,否则代码中的相对路径可能导致错误的发生
if __name__ == '__main__':
logging.config.dictConfig(LOGGING_CONFIG)