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

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