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.

Sign up for GitHub

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

Jump to bottom

Run at GPU #1

Open
ZFTurbo opened this issue Mar 7, 2022 · 1 comment
Open

Run at GPU #1

ZFTurbo opened this issue Mar 7, 2022 · 1 comment
Labels
good first issue Good for newcomers

Comments

@ZFTurbo
Copy link

ZFTurbo commented Mar 7, 2022

As I can see right now models run on CPU. Is it possible to run these models on GPU?
@yoyolicoris
Copy link
Owner

@ZFTurbo thanks for stopping by.
It's possible, but I probably won't implement it in the near future.
Currently, I'm more focusing on letting danna-sep be able to run on general devices as much as possible (e.g. converting to onnx format).
If you need GPU inference in a hurry, I would recommend you to make your own fork from v1.0.1, which still run all models in pytorch jit and can be modified easily.

@yoyolicoris yoyolicoris added the good first issue Good for newcomers label Mar 8, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
good first issue Good for newcomers
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@ZFTurbo @yoyolicoris