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

Onnx wrapper gpu,cpu support bug #444

Closed
sanjaypavo opened this issue May 7, 2022 · 2 comments
Closed

Onnx wrapper gpu,cpu support bug #444

sanjaypavo opened this issue May 7, 2022 · 2 comments
Assignees
@lvhan028

Comments

@sanjaypavo
Copy link
Contributor

Hi , I am facing an issue when trying to infer in onnx model in cpu. The problem is wrapper.py in this script in the line53, which sets the provider to CUDA if the device is CUDA enabled. I have debugged the error, willing to give a PR. Thanks in advance.
@lvhan028
Copy link
Collaborator

lvhan028 commented May 9, 2022

Look forward to your PR

@lvhan028 lvhan028 self-assigned this May 9, 2022
@lvhan028 lvhan028 reopened this May 11, 2022
@lvhan028
Copy link
Collaborator

@sanjaypavo Thanks very much for your PR (#532)

@sanjaypavo
Copy link
Contributor Author

@sanjaypavo Thanks very much for your PR (#532)

@lvhan028 Happy to contribute..

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@lvhan028 lvhan028

Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@lvhan028 @sanjaypavo