手动清理显存

photo_review/photo_review.py

@@ -2,6 +2,7 @@ import json
 import logging
 from time import sleep
 
+import paddle
 from sqlalchemy import update
 
 from config.keys import PATIENT_NAME, ADMISSION_DATE, DISCHARGE_DATE, MEDICAL_EXPENSES, PERSONAL_CASH_PAYMENT, \
@@ -214,6 +215,7 @@ def main():
                 session.execute(stmt)
                 session.commit()
                 session.close()
+                paddle.device.cuda.empty_cache()
         else:
             # 没有查询到新案子，等待一段时间后再查
             sleep_minutes = SLEEP_MINUTES