双卡配置
This commit is contained in:
3
main.py
3
main.py
@@ -1,4 +1,5 @@
|
|||||||
import logging.config
|
import logging.config
|
||||||
|
import os
|
||||||
import traceback
|
import traceback
|
||||||
|
|
||||||
from auto_email.error_email import send_an_error_email
|
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 config.photo_review import RETRY_TIME, SEND_ERROR_EMAIL
|
||||||
from photo_review.photo_review import main
|
from photo_review.photo_review import main
|
||||||
|
|
||||||
|
os.environ["CUDA_VISIBLE_DEVICES"] = "0,1"
|
||||||
|
|
||||||
# 项目必须从此处启动,否则代码中的相对路径可能导致错误的发生
|
# 项目必须从此处启动,否则代码中的相对路径可能导致错误的发生
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
logging.config.dictConfig(LOGGING_CONFIG)
|
logging.config.dictConfig(LOGGING_CONFIG)
|
||||||
|
|||||||
Reference in New Issue
Block a user