torch_npu 安装测试异常
收藏回复举报
torch_npu 安装测试异常
t('forum.solved') 已解决
新人帖
发表于2024-03-17 23:55:55
0 查看

安装完成运行测试异常

  • 硬件环境:opiaipro 16g
  • 系统版本:ubuntu 22.04 minimal aarch64
  • 软件版本:
    1. torch 版本 2.1.0
    2. torch-npu 版本 2.1.0
    3. CANN Ascend-cann-toolkit_7.0.0_linux-aarch64.run

按照安装手册操作完成后,测试异常。

HwHiAiUser@orangepiaipro:~/media$ python3 -c "import torch;import torch_npu;print(torch_npu.npu.is_available())"
/home/HwHiAiUser/.local/lib/python3.10/site-packages/torch_npu/dynamo/__init__.py:18: UserWarning: Register eager implementation for the 'npu' backend of dynamo, as torch_npu was not compiled with torchair.
  warnings.warn(
True

torch 和 torch-npu 版本

HwHiAiUser@orangepiaipro:~/media$ pip3 show torch
Name: torch
Version: 2.1.0
Summary: Tensors and Dynamic neural networks in Python with strong GPU acceleration
Home-page: https://pytorch.org/
Author: PyTorch Team
Author-email: packages@pytorch.org
License: BSD-3
Location: /home/HwHiAiUser/.local/lib/python3.10/site-packages
Requires: filelock, fsspec, jinja2, networkx, sympy, typing-extensions
Required-by: 
HwHiAiUser@orangepiaipro:~/media$ pip3 show torch-npu
Name: torch-npu
Version: 2.1.0
Summary: NPU bridge for PyTorch
Home-page: https://gitee.com/ascend/pytorch
Author: 
Author-email: 
License: BSD License
Location: /home/HwHiAiUser/.local/lib/python3.10/site-packages
Requires: 
Required-by:

脚步执行日志 版本

HwHiAiUser@orangepiaipro:~/media$ python3 -c "import torch;import torch_npu;torch.randn(2, 2).npu();"
/home/HwHiAiUser/.local/lib/python3.10/site-packages/torch_npu/dynamo/__init__.py:18: UserWarning: Register eager implementation for the 'npu' backend of dynamo, as torch_npu was not compiled with torchair.
  warnings.warn(
Traceback (most recent call last):
  File "<string>", line 1, in <module>
  File "/home/HwHiAiUser/.local/lib/python3.10/site-packages/torch/utils/backend_registration.py", line 153, in wrap_tensor_to
    device_idx = _normalization_device(custom_backend_name, device)
  File "/home/HwHiAiUser/.local/lib/python3.10/site-packages/torch/utils/backend_registration.py", line 109, in _normalization_device
    return _get_current_device_index()
  File "/home/HwHiAiUser/.local/lib/python3.10/site-packages/torch/utils/backend_registration.py", line 103, in _get_current_device_index
    return getattr(getattr(torch, custom_backend_name), _get_device_index)()
  File "/home/HwHiAiUser/.local/lib/python3.10/site-packages/torch_npu/npu/utils.py", line 51, in current_device
    torch_npu.npu._lazy_init()
  File "/home/HwHiAiUser/.local/lib/python3.10/site-packages/torch_npu/npu/__init__.py", line 203, in _lazy_init
    torch_npu._C._npu_init()
RuntimeError: Unsupported soc version: Ascend310B4

npu info

HwHiAiUser@orangepiaipro:~/media$ npu-smi info
+--------------------------------------------------------------------------------------------------------+
| npu-smi 23.0.0                                   Version: 23.0.0                                       |
+-------------------------------+-----------------+------------------------------------------------------+
| NPU     Name                  | Health          | Power(W)     Temp(C)           Hugepages-Usage(page) |
| Chip    Device                | Bus-Id          | AICore(%)    Memory-Usage(MB)                        |
+===============================+=================+======================================================+
| 0       310B4                 | Alarm           | 0.0          51                15    / 15            |
| 0       0                     | NA              | 0            2488 / 15609                            |
+===============================+=================+======================================================+

我要发帖子