Atlas 200I DK A2可以使用torch_npu的迁移功能,使用npu吗
收藏回复举报
Atlas 200I DK A2可以使用torch_npu的迁移功能,使用npu吗
t('forum.solved') 已解决
发表于2024-12-23 11:26:04
0 查看

环境:

python = 3.9.21
torch = 2.1.0
torch-npu = 2.1.0.post3
CANN = 7.0.RC1

我按照官方文档的教程安装了torch_npu,运行

python3 -c "import torch;import torch_npu; a = torch.randn(3, 4).npu(); print(a + a);"

报错:

EH9999: Inner Error!
EH9999  [Init][Env]init env failed![FUNC:ReportInnerError][FILE:log_inner.cpp][LINE:145]
        TraceBack (most recent call last):
        build op model failed, result = 500001[FUNC:ReportInnerError][FILE:log_inner.cpp][LINE:145]

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
RuntimeError: The Inner error is reported as above.
 Since the operator is called asynchronously, the stacktrace may be inaccurate. If you want to get the accurate stacktrace, pleace set the environment variable ASCEND_LAUNCH_BLOCKING=1.
[ERROR] 2024-12-23-11:03:43 (PID:40173, Device:0, RankID:-1) ERR00100 PTA call acl api failed

本帖最后由 匿名用户 于 2024/12/23 11:33:05 编辑

我要发帖子