tensor.to('npu')报错100002
收藏回复举报
tensor.to('npu')报错100002
t('forum.solved') 已解决
发表于2024-03-01 14:16:04
0 查看

import acl 

import torch 

import torch_npu 

import time 

 

acl.init() 

acl.rt.set_device(0) 

acl.mdl.load_from_file("./ocr_det2.om") 

device = torch.device('npu') 

x = torch.randn(1024,1024,dtype=torch.float16).to('npu') 

start_time_npu = time.time() 

result_npu = torch.mm(x, x) 

print(x.device) 

end_time_npu = time.time() 

print(f"NPU时间: {end_time_npu - start_time_npu}秒") 

acl初始化后,将张量转到npu上就会报错:

RuntimeError: Initialize:torch_npu/csrc/core/npu/sys_ctrl/npu_sys_ctrl.cpp:139 NPU error, error code is 100002 

[Error]: Repeated initialization or repeated loading. 

        Check whether the corresponding interface is repeatedly invoked for initialization or loading. 

报错初步看是重复初始化引起的,但我在acl初始化后,该如何将张量转到npu上呢?

我要发帖子