昇腾310,能在anaconda环境下使用torch-npu吗?
收藏回复举报
昇腾310,能在anaconda环境下使用torch-npu吗?
t('forum.solved') 已解决
新人帖
发表于2024-05-15 16:16:53
0 查看
在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 

我要发帖子