【MindSpore】CPU可以正常运行的，但是GPU下报错

最新推荐文章于 2024-03-19 15:31:50 发布

xiao | yang

最新推荐文章于 2024-03-19 15:31:50 发布

阅读量1.1k

点赞数

分类专栏：深度学习文章标签：深度学习人工智能机器学习

本文链接：https://blog.csdn.net/xi_xiyu/article/details/127611050

版权

在MindSpore中，当从CPU切换到GPU运行时遇到设备错误。尝试使用pynative动态图模式以获取更多错误信息，因为GPU可能不支持某些数据类型或导致loss为NAN。建议在关键点使用print()检查数据，特别是计算loss的地方，以定位问题。

摘要由CSDN通过智能技术生成

DEVICE 报错

【操作步骤&问题现象】

1、context.set_context(device_target="CPU") 正常跑

2、context.set_context(device_target="GPU") 报错

【截图信息】

【日志信息】（可选，上传日志内容或者附件）

[ERROR] DEVICE(3664,7f3ee8ff0700,python):2022-07-01-07:39:30.407.528 [mindspore/ccsrc/runtime/device/gpu/blocking_queue.cc:50] Push] Invalid Input: ptr: 0x7f3d702400a0, len: 262144
[ERROR] MD(3664,7f3ee8ff0700,python):2022-07-01-07:39:30.407.697 [mindspore/ccsrc/minddata/dataset/util/task_manager.cc:217] InterruptMaster] Task is terminated with err msg(more detail in info level log):Unexpected error. Invalid data, the types or shapes of current row is different with previous row(i.e. do batch operation but drop_reminder is False, or without resize image into the same size, these will cause shapes differs).
Line of code : 529
File         : /home/jenkins/agent