import torch 
import torch_npu 
def test_cpu(): 
    input = torch.randn(2000, 1000).detach().requires_grad_() 
    output = torch.sum(input) 
    output.backward(torch.ones_like(output)) 
def test_npu(): 
    input = torch.randn(2000, 1000).detach().requires_grad_().npu() 
    output = torch.sum(input) 
    output.backward(torch.ones_like(output)) 
if __name__ == "__main__": 
    test_cpu() 
    torch_npu.npu.set_device("npu:0") 
    test_npu()

(pytorch_npu) root@davinci-mini:/home # python Npu_test.py  

Traceback (most recent call last): 

  File "Npu_test.py", line 13, in <module> 

    torch_npu.npu.set_device("npu:0")  

  File "/usr/local/miniconda3/envs/pytorch_npu/lib/python3.8/site-packages/torch_npu/npu/utils.py", line 46, in set_device 

    torch_npu._C._npu_setDevice(device_id) 

RuntimeError: Unsupported soc version: Ascend310B4

npu-smi info ：npu-smi 23.0.rc2    Version: 23.0.rc2  

我的设备是Atlas 200I DK A2

按照教程首先安装了 Cann，

Ascend-cann-toolkit_7.0.0.alpha002_linux-aarch64.run

然后安装 2.1.0 的pytorch，

pip3 install torch==2.1.0

再安装torch-npu

pip3 install torch-npu==2.1.0rc1

最后运行脚本就会出现上方报错，是不是因为版本的问题呢，我尝试使用 Ascend-cann-toolkit_7.0.RC1_linux-aarch64.run，会出现同样报错，我应该如何解决这个问题，或者我应该安装什么版本的包