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

Update to Onnxruntime 1.2 and reenable its support for GPU #4919

Merged
merged 22 commits into from
Mar 10, 2020

Conversation

antoniovs1029
Copy link
Member

@antoniovs1029 antoniovs1029 commented Mar 4, 2020

Update dependencies in ML.NET for Onnxruntime.Managed 1.2 and reenable GPU support.

From now on OnnxTransformer will take a dependency on OnnxRuntime.Managed nuget, instead of OnnxRuntime. And users of ML.NET will have the ability to either use OnnxRuntime nuget (for CPU) or OnnxRuntime.Gpu nuget, depending if they want their onnx models to be applied using cpu or gpu. All of these, according to the changes that were suggested and implemented on microsoft/onnxruntime#2184

Notice that in # #4416 support for GPU was disabled because of changes in onnxruntime nugets, and in here that same code is reenabled.

@antoniovs1029 antoniovs1029 marked this pull request as ready for review March 10, 2020 22:38
@antoniovs1029 antoniovs1029 requested a review from a team as a code owner March 10, 2020 22:38
Copy link
Contributor

@harishsk harishsk left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

:shipit:

@antoniovs1029 antoniovs1029 changed the title [Draft] Update to Onnxruntime 1.2 and reenable its support for GPU Update to Onnxruntime 1.2 and reenable its support for GPU Mar 10, 2020
@antoniovs1029 antoniovs1029 merged commit c378772 into dotnet:master Mar 10, 2020
@ghost ghost locked as resolved and limited conversation to collaborators Mar 19, 2022
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants