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

yolov8pose类里没有cudafree? #256

Open
wcycqjy opened this issue Sep 30, 2024 · 2 comments
Open

yolov8pose类里没有cudafree? #256

wcycqjy opened this issue Sep 30, 2024 · 2 comments

Comments

@wcycqjy
Copy link

wcycqjy commented Sep 30, 2024

作者你好,我看网上cuda编程里都是分配内存然后清除内存,请问为什么在yolov8pose类里没有对变量的cudafree呢?

@wcycqjy
Copy link
Author

wcycqjy commented Oct 9, 2024

并且好像因为缺少对那几个host ptr device ptr的内存管理,持续的推理会导致内存占用累积出现内存溢出,在jetson上用jtop可以观察到。

@triple-Mu
Copy link
Owner

并且好像因为缺少对那几个host ptr device ptr的内存管理,持续的推理会导致内存占用累积出现内存溢出,在jetson上用jtop可以观察到。

https://github.com/triple-Mu/YOLOv8-TensorRT/blob/main/csrc/pose/normal/include/yolov8-pose.hpp#L108-L121
目前是在析构的时候清理的。是否可以定位到内存泄漏的位置呢?

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