调整指定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

View File

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