Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

python yolov9_trt.py erro #1493

Open
zmtttt opened this issue Apr 23, 2024 · 1 comment
Open

python yolov9_trt.py erro #1493

zmtttt opened this issue Apr 23, 2024 · 1 comment

Comments

@zmtttt
Copy link

zmtttt commented Apr 23, 2024

(base) zhaomt@gyu3nj-MS-7D54:~/com/tensorrtx/yolov9$ python yolov9_trt.py
[04/22/2024-17:58:10] [TRT] [I] Loaded engine size: 179 MiB
[04/22/2024-17:58:10] [TRT] [E] 1: [runtime.cpp::parsePlan::314] Error Code 1: Serialization (Serialization assertion plan->header.magicTag == rt::kPLAN_MAGIC_TAG failed.)
Traceback (most recent call last):
File "/home/zhaomt/com/tensorrtx/yolov9/yolov9_trt.py", line 435, in
yolov9_wrapper = yolov9TRT(engine_file_path)
^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/zhaomt/com/tensorrtx/yolov9/yolov9_trt.py", line 87, in init
context = engine.create_execution_context()
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
AttributeError: 'NoneType' object has no attribute 'create_execution_context'

PyCUDA ERROR: The context stack was not empty upon module cleanup.

A context was still active when the context stack was being
cleaned up. At this point in our execution, CUDA may already
have been deinitialized, so there is no way we can finish
cleanly. The program will be aborted now.
Use Context.pop() to avoid this problem.

Aborted (core dumped)

@WuxinrongY
Copy link
Contributor

AttributeError: 'NoneType' object has no attribute 'create_execution_context'产生的原因可能是engine没有成功建立,可以检查一下engine的路径。

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants