首页 新闻 会员 周边

pytorch训练出现out of memory

0
[待解决问题]

如何解决在用pytorch训练过程中，经过5 6个epoch训练速度变慢，GPU使用率达到100%，放到4090上运行还是出现相同的情况，加了torch.cuda.empty_cache()也不起作用

武、的主页 武、 | 菜鸟二级 | 园豆：202
提问于：2024-03-03 16:42
< >
分享
所有回答(2)
0

参考园子里的博文：pytorch学习笔记——训练时显存逐渐增加，几个epoch后out-of-memory

dudu | 园豆：30943 (高人七级) | 2024-03-03 20:03

多谢

支持(1) 反对(0) 武、 | 园豆：202 (菜鸟二级) | 2024-03-04 16:24
0

了解下混合精度的训练方式

zh-jp | 园豆：226 (菜鸟二级) | 2024-03-04 17:23
清除回答草稿
   您需要登录以后才能回答，未注册用户请先注册。