将GPU配置提取到配置文件中

photo_review/__init__.py

@@ -1,8 +1,12 @@
+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
+from config.photo_review import IE_BATCH_SIZE, LAYOUT_ANALYSIS, CUDA_VISIBLE_DEVICES
+
+os.environ["CUDA_VISIBLE_DEVICES"] = CUDA_VISIBLE_DEVICES
 
 # 基本医保结算单
 settlement_list_schema = \