Skip to content
New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

#

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

Jump to bottom

feat(gpu): add GPU support for model training purposes #497

Open
6 tasks
jkuri opened this issue Jan 31, 2021 · 0 comments
Open
6 tasks

feat(gpu): add GPU support for model training purposes #497

jkuri opened this issue Jan 31, 2021 · 0 comments
Assignees
@jkuri
Labels
Priority: High Status: Accepted Type: Enhancement

Comments

@jkuri
Copy link
Contributor

jkuri commented Jan 31, 2021 •
edited
Loading

We need to add support so abstruse can be configured that gpu availability comes into play for model training purposes.

  • worker detection for nvidia runtime
  • display that worker has GPU support on the UI
  • repository settings should include gpu and timeout option
  • repository should include options so you can limit resources for gpu, memory, ...
  • update scheduler to wait for gpu enabled worker in case of gpu enabled repo
  • prepare sample repo
# for free to join this conversation on GitHub. Already have an account? # to comment
Assignees

@jkuri jkuri

Labels
Priority: High Status: Accepted Type: Enhancement
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@jkuri