调整指定GPU
This commit is contained in:
10
main.py
10
main.py
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user