[发明专利]一种神经网络训练计算的迁移方法和装置有效
申请号: | 202010424387.9 | 申请日: | 2020-05-19 |
公开(公告)号: | CN111562988B | 公开(公告)日: | 2022-11-25 |
发明(设计)人: | 王萌 | 申请(专利权)人: | 苏州浪潮智能科技有限公司 |
主分类号: | G06F9/50 | 分类号: | G06F9/50;G06N3/08 |
代理公司: | 北京连和连知识产权代理有限公司 11278 | 代理人: | 刘小峰 |
地址: | 215100 江苏省苏州市吴*** | 国省代码: | 江苏;32 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | 本发明公开了一种神经网络训练计算的迁移方法和装置,该方法包括:编译透明调用脚本以获得使用统一内存分配接口替代显存分配接口的动态共享库;将动态共享库可调用地连接到统一内存分配接口和显存分配接口以加载到环境变量中;使用内核未经修改的深度学习框架连接到显存分配接口并在环境变量中训练神经网络结构;响应于训练神经网络结构所需的显存超过最大显存的第一阈值,而使用动态共享库将上层页面通过统一内存分配接口和显存分配接口以页错误和页迁移的方式迁移到统一内存以执行训练。本发明能够透明修改PyTorch的内存分配并整体迁移上层页面,提高内存管理灵活性和迁移计算量,使得神经网络训练更方便更有效率。 | ||
搜索关键词: | 一种 神经网络 训练 计算 迁移 方法 装置 | ||
【主权项】:
暂无信息
下载完整专利技术内容需要扣除积分,VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于苏州浪潮智能科技有限公司,未经苏州浪潮智能科技有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/patent/202010424387.9/,转载请声明来源钻瓜专利网。
- 上一篇:角柜配件及角柜
- 下一篇:一种高性能耐化学腐蚀胶带