Halcon视觉之家-51Halcon专注于机器视觉技术

 找回密码
 会员注册

QQ登录

只需一步,快速开始

扫一扫,微信登录

查看: 347|回复: 8

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

[复制链接]
  • TA的每日心情
    开心
    7 天前
  • 签到天数: 33 天

    连续签到: 1 天

    [LV.5]常住居民I

    18

    主题

    97

    帖子

    175

    积分

    Rank: 1

    积分
    175

    切换助手验证会员

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

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

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

    x
    同事们,早上好!

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

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

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

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

  • TA的每日心情
    郁闷
    2022-4-7 15:38
  • 签到天数: 4 天

    连续签到: 2 天

    [LV.2]偶尔看看I

    7

    主题

    34

    帖子

    73

    积分

    Rank: 1

    积分
    73
    发表于 2022-2-13 16:39:53 | 显示全部楼层
    缩小batch size
  • TA的每日心情
    开心
    7 天前
  • 签到天数: 33 天

    连续签到: 1 天

    [LV.5]常住居民I

    18

    主题

    97

    帖子

    175

    积分

    Rank: 1

    积分
    175

    切换助手验证会员

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

    不可能再低了)))。
  • TA的每日心情
    开心
    7 天前
  • 签到天数: 33 天

    连续签到: 1 天

    [LV.5]常住居民I

    18

    主题

    97

    帖子

    175

    积分

    Rank: 1

    积分
    175

    切换助手验证会员

     楼主| 发表于 2022-2-14 13:38:00 | 显示全部楼层
    不可能再低了)))。
  • TA的每日心情
    奋斗
    昨天 08:30
  • 签到天数: 633 天

    连续签到: 2 天

    [LV.9]以坛为家II

    0

    主题

    139

    帖子

    851

    积分

    Rank: 3

    积分
    851

    切换助手验证会员

    发表于 2022-2-15 13:00:40 | 显示全部楼层
    哪一个版本?
  • TA的每日心情
    擦汗
    2022-2-21 08:43
  • 签到天数: 7 天

    连续签到: 1 天

    [LV.3]偶尔看看II

    2

    主题

    14

    帖子

    35

    积分

    Rank: 1

    积分
    35

    切换助手验证会员

    发表于 2022-2-15 13:56:57 | 显示全部楼层
    在read_dl_model之前加入这段代码试试
    set_system ('cudnn_deterministic', 'true')。
  • TA的每日心情
    开心
    2022-4-6 21:12
  • 签到天数: 65 天

    连续签到: 1 天

    [LV.6]常住居民II

    0

    主题

    56

    帖子

    152

    积分

    Rank: 1

    积分
    152
    发表于 2022-2-16 09:09:55 | 显示全部楼层
    let me mark
  • TA的每日心情
    开心
    7 天前
  • 签到天数: 33 天

    连续签到: 1 天

    [LV.5]常住居民I

    18

    主题

    97

    帖子

    175

    积分

    Rank: 1

    积分
    175

    切换助手验证会员

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

    谢谢大家的答复。

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

    我在其中训练神经网络模型的Halcon版本 20.11.
  • TA的每日心情
    开心
    7 天前
  • 签到天数: 33 天

    连续签到: 1 天

    [LV.5]常住居民I

    18

    主题

    97

    帖子

    175

    积分

    Rank: 1

    积分
    175

    切换助手验证会员

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

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

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

    本版积分规则

    视觉培训招生

    建议您使用Chrome、Firefox、Edge、IE10及以上版本和360等主流浏览器浏览本网站

    51Halcon会员技术交流会员技术交流 | 51Halcon官方客服咨询官方客服咨询 | Halcon切换助手使用反馈切换助手使用

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

    © 2015-2021 51Halcon机器视觉  X3.4  粤ICP备15095995号 粤公网安备44030602000670号

    本站已安全运行:

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