PP OCR4 GPU训练内存溢出问题如何解决

项目

数据集

课程

比赛

模型库

活动

论坛

访问飞桨官网

项目

数据集

课程

比赛

模型库

活动

论坛

访问飞桨官网

M Merious浅末发布于2023-09

使用PP-OCR4 GPU进行训练时 gpu直接飙升至100%导致失败

调整batch_size及work_nums 均无法解决问题

PP-OCR3 可以正常训练

0

收藏

回复

全部评论(1)

宇宙物语

#2 回复于2023-09

使用具有更大显存的GPU

0

回复

提issue

需求/bug反馈？一键提issue告诉我们

提pr

发现bug？如果您知道修复办法，欢迎提pr直接参与建设飞桨~