pytorch迁移后,张量运算结果有问题
收藏回复举报
pytorch迁移后,张量运算结果有问题
t('forum.solved') 已解决
发表于2024-07-09 14:09:31
0 查看

大家好,

我在使用晟腾 NPU 进行张量运算时遇到了一些问题。配置如下:

  • 主机架构: aarch64
  • CANN 版本: 7.0.0
  • PyTorch 版本: 2.1.0
  • 驱动与固件版本: 24.1

使用的是pytorch自动迁移。

下面是我的测试代码:

执行的是一个简单的张量加法操作,但是结果却不正确。

下面是代码运行结果:

直接设置成npu也还是不行:

服务器信息如下:

请问有没有同学遇到过类似的问题?该如何解决?谢谢大家的帮助!

我要发帖子