开发者社区 > ModelScope模型即服务 > 正文

发生cuda设备丢失错误, 没有任何其他警告. 是否是ModelScope训练代码的问题?

在新的dataset方式中测试, 如果用#号指定的抽样数量大于数据集的条数, 会在训练中直接崩掉gpu, 发生cuda设备丢失错误, 没有任何其他警告. 是否是ModelScope训练代码的问题? 如果数据集中抽样确实不能大于样本总量 也就是不会重复抽取, 是否添加一个报错会比较合适? Unable to determine the device handle for GPU 0000:42:00.0: GPU is lost. Reboot the system to recover this GPU

展开
收起
小小爱吃香菜 2024-05-21 21:10:05 52 0
0 条回答
写回答
取消 提交回答

ModelScope旨在打造下一代开源的模型即服务共享平台,为泛AI开发者提供灵活、易用、低成本的一站式模型服务产品,让模型应用更简单!欢迎加入技术交流群:微信公众号:魔搭ModelScope社区,钉钉群号:44837352

相关电子书

更多
视觉AI能力的开放现状及ModelScope实战 立即下载
ModelScope助力语音AI模型创新与应用 立即下载
低代码开发师(初级)实战教程 立即下载