Halcon视觉之家-Halcon视觉技术论坛

 找回密码
 会员注册

QQ登录

只需一步,快速开始

扫一扫,微信登录

查看: 563|回复: 8

[HalconDL] 问题是在GPU上学习模型

[复制链接]

签到天数: 6 天

连续签到: 1 天

[LV.2]偶尔看看I

18

主题

103

帖子

204

积分

Rank: 2Rank: 2

积分
204

切换助手验证会员

发表于 2022-2-11 13:08:08 | 显示全部楼层 |阅读模式

51Halcon诚邀您的加入,专注于机器视觉开发与应用技术,我们一直都在努力!

您需要 登录 才可以下载或查看,没有账号?会员注册

x
同事们,早上好!

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

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

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

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

该用户从未签到

7

主题

35

帖子

86

积分

Rank: 1

积分
86
发表于 2022-2-13 16:39:53 | 显示全部楼层
缩小batch size

签到天数: 6 天

连续签到: 1 天

[LV.2]偶尔看看I

18

主题

103

帖子

204

积分

Rank: 2Rank: 2

积分
204

切换助手验证会员

 楼主| 发表于 2022-2-14 13:31:55 | 显示全部楼层
BatchSize=1

不可能再低了)))。

签到天数: 6 天

连续签到: 1 天

[LV.2]偶尔看看I

18

主题

103

帖子

204

积分

Rank: 2Rank: 2

积分
204

切换助手验证会员

 楼主| 发表于 2022-2-14 13:38:00 | 显示全部楼层
不可能再低了)))。

签到天数: 66 天

连续签到: 3 天

[LV.6]常住居民II

0

主题

142

帖子

1012

积分

Rank: 4Rank: 4

积分
1012

切换助手验证会员

发表于 2022-2-15 13:00:40 | 显示全部楼层
哪一个版本?

该用户从未签到

2

主题

14

帖子

36

积分

Rank: 1

积分
36

切换助手验证会员

发表于 2022-2-15 13:56:57 | 显示全部楼层
在read_dl_model之前加入这段代码试试
set_system ('cudnn_deterministic', 'true')。

签到天数: 6 天

连续签到: 1 天

[LV.2]偶尔看看I

0

主题

62

帖子

176

积分

Rank: 1

积分
176
发表于 2022-2-16 09:09:55 | 显示全部楼层
let me mark

签到天数: 6 天

连续签到: 1 天

[LV.2]偶尔看看I

18

主题

103

帖子

204

积分

Rank: 2Rank: 2

积分
204

切换助手验证会员

 楼主| 发表于 2022-2-16 16:42:31 | 显示全部楼层
大家下午好。

谢谢大家的答复。

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

我在其中训练神经网络模型的Halcon版本 20.11.

签到天数: 6 天

连续签到: 1 天

[LV.2]偶尔看看I

18

主题

103

帖子

204

积分

Rank: 2Rank: 2

积分
204

切换助手验证会员

 楼主| 发表于 2022-2-17 14:59:57 | 显示全部楼层
同事们,大家好。

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

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

本版积分规则

视觉培训招生
论坛下线通知X

建议您使用Chrome、Firefox、Edge、360、IE10以上等主流浏览器浏览本论坛

算子查询| 申请友链| 小黑屋| 手机版| Archiver| 业务支持|

© 2015-2021 Halcon视觉之家  X3.4  粤ICP备15095995号 粤公网安备44030602000670号

本站已安全运行:

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