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

yolo11s-pose.pt 模型使用trt加速和不使用trt加速的耗时问题 #264

Open
stephen-TT opened this issue Nov 6, 2024 · 1 comment

Comments

@stephen-TT
Copy link

1050ti;
cuda11.4
pytorch: 1.12.1
tensorrt:TensorRT-8.5.3.1
model: yolo11s-pose.pt

pytorch安装方式:conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch

使用tensorrt加速耗时在29ms,并且还是用fp16格式导出的engine,使用官方pt耗时在24ms。

为什么在相同的机器上,使用trt加速反而变慢了。

望回复

@triple-Mu
Copy link
Owner

如何计算的时间呢?以及是否使用了end2end

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