ATC转化yolov5模型失败
收藏回复举报
ATC转化yolov5模型失败
t('forum.solved') 已解决
新人帖
发表于2025-02-28 11:15:10
0 查看
我自己训练的官方yolov5模型,环境是物理机ubuntu22.04+cuda11.8+pytorch2.3.1,转换的onnx模型无误
然后使用指南中在物理机ubuntu中ATC转换模型失败,命令:atc --model=runs/train/exp/weights/best.onnx --framework=5 --output=runs/train/exp/weights --input_format=NCHW --soc_version=Ascend310B4 --input_shape="images:1,3,640,640"
日志如下:
 

E90000: 2025-02-28-11:07:48.938.436 Compile operator failed, cause: module 'sys' has no attribute 'argv'   File "/home/sd/Ascend/ascend-toolkit/latest/python/site-packages/tbe/common/repository_manager/interface.py", line 33, in cann_kb_init 

    return RouteServer.initialize(**locals()) 

  File "/home/sd/Ascend/ascend-toolkit/latest/python/site-packages/tbe/common/repository_manager/route.py", line 53, in wrapper 

    return func(cls, *args, **kwargs) 

  File "/home/sd/Ascend/ascend-toolkit/latest/python/site-packages/tbe/common/repository_manager/route.py", line 171, in initialize 

    process_item.start() 

  File "/home/sd/anaconda3/lib/python3.7/multiprocessing/process.py", line 112, in start 

    self._popen = self._Popen(self) 

  File "/home/sd/anaconda3/lib/python3.7/multiprocessing/context.py", line 291, in _Popen 

    return Popen(process_obj) 

  File "/home/sd/anaconda3/lib/python3.7/multiprocessing/popen_forkserver.py", line 35, in __init__ 

    super().__init__(process_obj) 

  File "/home/sd/anaconda3/lib/python3.7/multiprocessing/popen_fork.py", line 20, in __init__ 

    self._launch(process_obj) 

  File "/home/sd/anaconda3/lib/python3.7/multiprocessing/popen_forkserver.py", line 42, in _launch 

    prep_data = spawn.get_preparation_data(process_obj._name) 

  File "/home/sd/anaconda3/lib/python3.7/multiprocessing/spawn.py", line 163, in get_preparation_data 

    sys_argv=sys.argv, 

 

        TraceBack (most recent call last): 

        AOE Failed to call InitCannKB 

        [GraphOpt][InitializeInner][InitTbeFunc] Failed to init tbe.[FUNC:InitializeInner][FILE:tbe_op_store_adapter.cc][LINE:1719] 

        [SubGraphOpt][PreCompileOp][InitAdapter] InitializeAdapter adapter [tbe_op_adapter] failed! Ret [4294967295][FUNC:InitializeAdapter][FILE:op_store_adapter_manager.cc][LINE:79] 

        [SubGraphOpt][PreCompileOp][Init] Initialize op store adapter failed, OpsStoreName[tbe-custom].[FUNC:Initialize][FILE:op_store_adapter_manager.cc][LINE:120] 

        [FusionMngr][Init] Op store adapter manager init failed.[FUNC:Initialize][FILE:fusion_manager.cc][LINE:117] 

        PluginManager InvokeAll failed.[FUNC:Initialize][FILE:ops_kernel_manager.cc][LINE:82] 

        OpsManager initialize failed.[FUNC:InnerInitialize][FILE:gelib.cc][LINE:234] 

        GELib::InnerInitialize failed.[FUNC:Initialize][FILE:gelib.cc][LINE:162] 

我要发帖子