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

[FEAT][CV] Add support for GPU objects #787

Closed
noamzbr opened this issue Jan 31, 2022 · 0 comments · Fixed by #832
Closed

[FEAT][CV] Add support for GPU objects #787

noamzbr opened this issue Jan 31, 2022 · 0 comments · Fixed by #832
Assignees

Comments

@noamzbr
Copy link
Collaborator

noamzbr commented Jan 31, 2022

Is your feature request related to a problem? Please describe.
Current vision code supports only cpu objects, and even if user supplies a GPU model or data it'll crash.

Describe the solution you'd like
Need a global object (context) to recognize both if GPU is available and if the user objects are GPU, and adjust the rest of the code to cope with that.

@noamzbr noamzbr added this to the computer-vision milestone Jan 31, 2022
@noamzbr noamzbr changed the title [FEAT][CV] [FEAT][CV] Add support for GPU objects Jan 31, 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 a pull request may close this issue.

2 participants