设为首页
收藏本站
最新授权:2023.02
切换到宽版
登录
注册
找回密码
搜索
搜索
本版
帖子
用户
快捷导航
首页
门户首页
论坛
论坛首页
授权
Halcon试用授权下载
Halcon授权
Halcon视觉技术网
»
论坛
›
视觉软件技术
›
Halcon技术
›
问题是在GPU上学习模型
返回列表
问题是在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
|
显示全部楼层
同事们,大家好。
不幸的是,所有的选项都不起作用。
最后,我不得不把硬盘的体积添加到交换文件中。不幸的是,这是唯一的办法。
回复
举报
返回列表
高级模式
B
Color
Image
Link
Quote
Code
Smilies
您需要登录后才可以回帖
登录
|
注册
本版积分规则
发表回复
回帖后跳转到最后一页
快速回复
返回顶部
返回列表