专栏算法工具链horizon_plugin_pytorch哪里下载？

horizon_plugin_pytorch哪里下载？

已解决
jalymo2023-04-25
95
7

请问：horizon_plugin_pytorch哪里下载到？

ot@f40f1c19bda5:/open_explorer/horizon_model_convert_sample/ModelZoo/customer# python customer_train.py

Traceback (most recent call last):

File "customer_train.py", line 9, in <module>

import horizon_plugin_pytorch as horizon

ModuleNotFoundError: No module named 'horizon_plugin_pytorch'

算法工具链
评论3
0/1000
  • 颜值即正义
    Lv.2

    你好，使用plugin的话，建议使用GPU docker。

    然后直接答复你的问题，horizon_plugin_pytorch.whl在OE包路径ddk/package/host/ai_toolchain/horizon_plugin_pytorch-1.4.4+cu111.torch1102-cp38-cp38-linux_x86_64.whl
    2023-04-26
    0
    2
    • jalymo回复颜值即正义:

      我本机没有cuda和gpu卡，纯cpu的也可以用这个horizon plugin pytorch吗？

      2023-04-27
      0
    • 颜值即正义回复jalymo:

      纯CPU不太行哈

      2023-04-27
      0
  • jalymo
    Lv.1
    您好，请问：bev_release_package-1.6.29/bev_release_package一定要用gpu吗？我本地只有cpu而且也是下载的cpu镜像docker_openexplorer_ubuntu_20_j5_cpu_v1.1.52a.tar.gz，没有运行报错如下：bev_release_package-1.6.29/bev_release_package# python3 tools/train.py

    Traceback (most recent call last):

    File "/usr/local/lib/python3.8/dist-packages/horizon_plugin_pytorch/extension.py", line 35, in _register_extensions

    torch.ops.load_library(ext_specs.origin)

    File "/root/.local/lib/python3.8/site-packages/torch/_ops.py", line 110, in load_library

    ctypes.CDLL(path)

    File "/usr/lib/python3.8/ctypes/__init__.py", line 373, in __init__

    self._handle = _dlopen(self._name, mode)

    OSError: libcuda.so.1: cannot open shared object file: No such file or directory

    During handling of the above exception, another exception occurred:

    Traceback (most recent call last):

    File "tools/train.py", line 10, in

    import horizon_plugin_pytorch as horizon

    File "/usr/local/lib/python3.8/dist-packages/horizon_plugin_pytorch/__init__.py", line 8, in

    from . import extension, jit, nn, quantization # noqa: F401

    File "/usr/local/lib/python3.8/dist-packages/horizon_plugin_pytorch/extension.py", line 42, in

    _register_extensions()

    File "/usr/local/lib/python3.8/dist-packages/horizon_plugin_pytorch/extension.py", line 37, in _register_extensions

    raise RuntimeError(

    RuntimeError: Fail to load libhorizon_ops, origin msg is

    libcuda.so.1: cannot open shared object file: No such file or directory

    2023-06-10
    0
    2
    • 颜值即正义回复jalymo:

      你好，训练用CPU不ok的，需要用GPU~

      2023-06-10
      0
    • jalymo回复颜值即正义:

      您好，我现在要把模型转成Device端运行的也不行，请问：cpu docker如何进行bev模型转换？

      log如下：

      bev_release_package-1.6.29# python3 tools/compile_perf.py --config configs/bev/bev_mt_ipm.py --out-dir ./ --opt 3

      Traceback (most recent call last):

      File "/usr/local/lib/python3.8/dist-packages/horizon_plugin_pytorch/extension.py", line 35, in _register_extensions

      torch.ops.load_library(ext_specs.origin)

      File "/root/.local/lib/python3.8/site-packages/torch/_ops.py", line 110, in load_library

      ctypes.CDLL(path)

      File "/usr/lib/python3.8/ctypes/__init__.py", line 373, in __init__

      self._handle = _dlopen(self._name, mode)

      OSError: libcuda.so.1: cannot open shared object file: No such file or directory

      During handling of the above exception, another exception occurred:

      Traceback (most recent call last):

      File "tools/compile_perf.py", line 10, in

      import horizon_plugin_pytorch as horizon

      File "/usr/local/lib/python3.8/dist-packages/horizon_plugin_pytorch/__init__.py", line 8, in

      from . import extension, jit, nn, quantization # noqa: F401

      File "/usr/local/lib/python3.8/dist-packages/horizon_plugin_pytorch/extension.py", line 42, in

      _register_extensions()

      File "/usr/local/lib/python3.8/dist-packages/horizon_plugin_pytorch/extension.py", line 37, in _register_extensions

      raise RuntimeError(

      RuntimeError: Fail to load libhorizon_ops, origin msg is

      libcuda.so.1: cannot open shared object file: No such file or directory

      2023-06-10
      0
  • 颜值即正义
    Lv.2
    2023-04-26
    0
    0