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

FPS in 1080ti just 30-50fps for single_class detection #33

Closed
xxy90 opened this issue Nov 27, 2020 · 1 comment
Closed

FPS in 1080ti just 30-50fps for single_class detection #33

xxy90 opened this issue Nov 27, 2020 · 1 comment

Comments

@xxy90
Copy link

xxy90 commented Nov 27, 2020

image

@RangiLyu
Copy link
Owner

RangiLyu commented Nov 27, 2020

Make sure you turn on cudnn acceleration.
Pull the latest version of nanodet or change

    torch.backends.cudnn.deterministic = True
    torch.backends.cudnn.benchmark = False

in demo.py to

    torch.backends.cudnn.enabled = True
    torch.backends.cudnn.benchmark = True

By the way, pytorch is a training framework, not designed for inference. If you want to get better inference speed on GPU, try to use C++ demo. C++ demo using ncnn vulkan as backend.

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