首页 Paddle框架 帖子详情
训练时报错提示显存不足,如何解决? 已解决
收藏
快速回复
Paddle框架 问答模型训练深度学习 1054 1
训练时报错提示显存不足,如何解决? 已解决
收藏
快速回复
Paddle框架 问答模型训练深度学习 1054 1

训练时报错提示显存不足,如何解决?

旧鸱c_
已解决
2# 回复于2021-08
可以尝试按如下方法解决: 检查是当前模型是否占用了过多显存,可尝试减小batch_size ; 开启以下三个选项: #一旦不再使用即释放内存垃圾,=1.0 垃圾占用内存大小达到10G时,释放内存垃圾 export FLAGS_eager_delete_tensor_gb=0.0 #启用快速垃圾回收策略,不等待cuda kernel 结束,直接释放显存 export FLAGS_fast_eager_deletion_mode=1 #该环境变量设置只占用0%的显存 export FLAGS_fraction_of_gpu_memory_to_use=0   详细请参考官方文档存储分配与优化 调整相关配置。 此外,建议您使用AI Studio 学习与 实训社区训练,获取免费GPU算力,提升您的训练效率。
展开
0
收藏
回复
全部评论(1)
时间顺序
旧鸱c_
#2 回复于2021-08

可以尝试按如下方法解决:
检查是当前模型是否占用了过多显存,可尝试减小batch_size ;
开启以下三个选项:
#一旦不再使用即释放内存垃圾,=1.0 垃圾占用内存大小达到10G时,释放内存垃圾
export FLAGS_eager_delete_tensor_gb=0.0
#启用快速垃圾回收策略,不等待cuda kernel 结束,直接释放显存
export FLAGS_fast_eager_deletion_mode=1
#该环境变量设置只占用0%的显存
export FLAGS_fraction_of_gpu_memory_to_use=0

 

详细请参考官方文档存储分配与优化 调整相关配置。

此外,建议您使用AI Studio 学习与 实训社区训练,获取免费GPU算力,提升您的训练效率。

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户