昇腾论坛

torch-npu不支持？

已解决

发表于2024-09-27 15:15:11

0 查看

按https://ascend.github.io/docs/sources/pytorch/index.html安装torch-npu没有报错，但是运行示例代码报错

示例代码：

import torch
import torch_npu

x = torch.randn(2, 2).npu()
y = torch.randn(2, 2).npu()
z = x.mm(y)

print(z)

匿名回复

我要发帖子