MindSpore报错TypeError: Can not select a valid kernel info for [Default/Pow-op0] in AI CORE or AI CPU

本文介绍了在使用MindSpore 1.6.0版本时遇到的TypeError,该错误发生在GPU环境下尝试使用Pow操作符。错误原因是传入的数据类型为float64,而MindSpore仅支持int8、uint8、int32、float16、float32。解决方法是将数据类型转换为支持的类型。文章提供了报错的脚本段落和解决步骤。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

1 报错描述

1.1 系统环境

Hardware Environment(Ascend/GPU/CPU): GPU

Software Environment:

– MindSpore version (source or binary): 1.6.0

– Python version (e.g., Python 3.7.5): 3.7.6

– OS platform and distribution (e.g., Linux Ubuntu 16.04): Ubuntu 4.15.0-74-generic

– GCC/Compiler version (if compiled from source):

1.2 基本信息

1.2.1 脚本

训练脚本是通过构建Pow的单算子网络,对第一个输入的每个值进行幂运算。脚本如下:

 01 class Net(nn.Cell):
 02   def __init__(self):
 03     super(Net, self).__init__()
 04     self.pow = ops.Pow()
 05   def construct(self, x,y):
 06     output = self.pow(x, y)
 07     return output
 08 net = Net()
 09 x = Tensor(np.array
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值