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(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

Comments

@jkuri
Copy link
Contributor

jkuri commented Jan 31, 2021

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
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

1 participant