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

How to Run Python API module with Onnxruntime gpu (Cuda) #6

Open
7sn-The-Dreamer opened this issue Apr 13, 2024 · 1 comment
Open

Comments

@7sn-The-Dreamer
Copy link

First of all, thank you so much for this outstanding work.

My question is, after pip install the prebuilt python module, how to run the Python api with Onnxruntime-gpu (CUDA) to get use of the GPU acceleration?

Thank you so much 🙏🏼.

@sergei-alabugin
Copy link
Collaborator

Hi @7sn-The-Dreamer !
Thank for you interest in our repo.
Unfortunately, there is no possibility to use our open-source SDK with GPU.
I suggest you to consider 2 options:

  1. Our commercial SDK https://3divi.ai/products/software/face-sdk which have much more features including GPU support
  2. You can replace libonnxruntime.so which used for project building (it's downloaded during build from source) by libonnxruntime from here https://github.com/microsoft/onnxruntime/releases

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