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

add yolov5 serverless function leveraging nvidia gpu #4960

Merged
merged 1 commit into from
Oct 12, 2022

Conversation

tdhooghe
Copy link
Contributor

@tdhooghe tdhooghe commented Sep 16, 2022

Fixes #4851

This change enables leveraging an NVIDIA GPU when using a YOLOv5 serverless function.

Motivation and context

This change speeds up auto labeling with a YOLOv5 model.

How has this been tested?

Deploying the script using deploy_gpu.sh and checking if it uses the NVIDIA GPU with NVIDIA-SMI while auto-labeling.

Checklist

License

  • I submit my code changes under the same MIT License that covers the project.
    Feel free to contact the maintainers if that's a concern.

@bsekachev bsekachev self-assigned this Oct 10, 2022
@azhavoro
Copy link
Contributor

LGTM, worked for me

@azhavoro
Copy link
Contributor

@tdhooghe Hi,
Thanks for the your contribution and could you add a line to the changelog about your change?

@bsekachev bsekachev merged commit 5936bc5 into cvat-ai:develop Oct 12, 2022
@bsekachev
Copy link
Member

@tdhooghe

Thank you for the contribution, I will add a line to CONTRIBUTING.md later.

@tdhooghe
Copy link
Contributor Author

tdhooghe commented Oct 13, 2022 via email

@bsekachev
Copy link
Member

@tdhooghe
Would be great, thank you.

@nmanovic nmanovic mentioned this pull request Dec 12, 2022
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

Successfully merging this pull request may close these issues.

Unable to load the YOLOv5 on to the GPU
3 participants