开启左侧

问题是在GPU上学习模型

[复制链接]
Evgeniy 发表于 2022-2-11 13:08:08 | 显示全部楼层 |阅读模式
同事们,早上好!

你能告诉我如何在训练神经网络模型时减少RAM的负荷吗?

简单的例子dl_deteciion_workflow.hdev,当在GPU上训练时,它吃掉了所有的内存,进程崩溃了。显示一个内存溢出的错误。

我的电脑是英特尔i5--第十代,12个核心,16G内存,NVidia GeForce 1650 Ti。

最有趣的是,在CPU上训练时,这个问题并没有发生!!。

进击的烤肉 发表于 2022-2-13 16:39:53 | 显示全部楼层
缩小batch size
 楼主| Evgeniy 发表于 2022-2-14 13:31:55 | 显示全部楼层
BatchSize=1

不可能再低了)))。
朱颜改 发表于 2022-2-15 13:56:57 | 显示全部楼层
在read_dl_model之前加入这段代码试试
set_system ('cudnn_deterministic', 'true')。
xuyongze 发表于 2022-2-16 09:09:55 | 显示全部楼层
let me mark {:6_192:}{:6_192:}
 楼主| Evgeniy 发表于 2022-2-16 16:42:31 | 显示全部楼层
大家下午好。

谢谢大家的答复。

我现在就去试试,并写下我的结果。

我在其中训练神经网络模型的Halcon版本 20.11.
 楼主| Evgeniy 发表于 2022-2-17 14:59:57 | 显示全部楼层
同事们,大家好。

不幸的是,所有的选项都不起作用。

最后,我不得不把硬盘的体积添加到交换文件中。不幸的是,这是唯一的办法。
您需要登录后才可以回帖 登录 | 注册

本版积分规则

快速回复 返回顶部 返回列表