我自己训练的官方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]
然后使用指南中在物理机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]