在usr下安装了Anaconda和CANN7.0.1,设置了python3.8的环境,然后下载了对应的torch和torch-npu,然后测试了一下:
import torch
import torch_npu
x = torch.randn(2, 2).npu()
现在显示 RuntimeError: Unsupported soc version: Ascend310
具体log:
Traceback (most recent call last):
File "test.py", line 4, in <module>
x = torch.randn(2, 2).npu()
File "/home/storage/.conda/envs/py38/lib/python3.8/site-packages/torch_npu/utils/device_guard.py", line 45, in wrapper
return func(*args, **kwargs)
File "/home/storage/.conda/envs/py38/lib/python3.8/site-packages/torch_npu/utils/tensor_methods.py", line 72, in _npu
return torch_npu._C.npu(self, *args, **kwargs)
RuntimeError: Unsupported soc version: Ascend310
import torch
import torch_npu
x = torch.randn(2, 2).npu()
现在显示 RuntimeError: Unsupported soc version: Ascend310
具体log:
Traceback (most recent call last):
File "test.py", line 4, in <module>
x = torch.randn(2, 2).npu()
File "/home/storage/.conda/envs/py38/lib/python3.8/site-packages/torch_npu/utils/device_guard.py", line 45, in wrapper
return func(*args, **kwargs)
File "/home/storage/.conda/envs/py38/lib/python3.8/site-packages/torch_npu/utils/tensor_methods.py", line 72, in _npu
return torch_npu._C.npu(self, *args, **kwargs)
RuntimeError: Unsupported soc version: Ascend310