调整指定GPU

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)

photo_review/__init__.py

@@ -1,12 +1,8 @@
-import os
-
 from paddlenlp import Taskflow
 
 from config.keys import PATIENT_NAME, ADMISSION_DATE, DISCHARGE_DATE, MEDICAL_EXPENSES, PERSONAL_CASH_PAYMENT, \
     PERSONAL_ACCOUNT_PAYMENT, PERSONAL_FUNDED_AMOUNT, MEDICAL_INSURANCE_TYPE, HOSPITAL, DEPARTMENT, DOCTOR
-from config.photo_review import IE_BATCH_SIZE, LAYOUT_ANALYSIS, CUDA_VISIBLE_DEVICES
+from config.photo_review import IE_BATCH_SIZE, LAYOUT_ANALYSIS
-
-os.environ["CUDA_VISIBLE_DEVICES"] = CUDA_VISIBLE_DEVICES
 
 # 基本医保结算单
 settlement_list_schema = \