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

Does gonnx supports GPU inference? #206

Open
deependujha opened this issue Sep 21, 2024 · 1 comment
Open

Does gonnx supports GPU inference? #206

deependujha opened this issue Sep 21, 2024 · 1 comment

Comments

@deependujha
Copy link

As stated in the heading, is there a way to infer model on GPUs or TPUs?

@Swopper050
Copy link
Collaborator

Hi @deependujha , right now it is not supported out of the box in Gonnx. However, I believe that the Tensor package we use does support GPU inference so we will look into building this in the future (or feel free to try yourself).

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

No branches or pull requests

2 participants