-
Notifications
You must be signed in to change notification settings - Fork 99
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
Onnxruntime v1.8.1 with GPU/CUDA 11 support #87
base: master
Are you sure you want to change the base?
Conversation
Thanks for the contribution! Sorry for my late replay, I don't have much time to put on that project... I updated the bindings yesterday to 1.8.1. Would you mind pushing commits just for the CUDA stuff? Or I can cherry-pick the two if you prefer. |
a4a688d
to
be04fbe
Compare
All set! I have rebase from master ! I did not push the bindings for Windows as I have no machine to test it on. Feel free to leave comment on necessary modification :) |
be04fbe
to
61f05c5
Compare
Codecov Report
@@ Coverage Diff @@
## master #87 +/- ##
==========================================
- Coverage 16.11% 14.40% -1.71%
==========================================
Files 19 20 +1
Lines 1117 1277 +160
==========================================
+ Hits 180 184 +4
- Misses 937 1093 +156
Continue to review full report at Codecov.
|
fix apple gpu download bug adding windows bindings fixing apple no gpu problem fixing workflow mistake
7578dc0
to
045e470
Compare
This branch should pass test now :) |
Merging two branches together to enables CUDA 11 Support on ONNX 1.7
The first one is: #78
The second one is: https://github.com/radu-matei/onnxruntime-rs/tree/cuda
This solution may not be perfect but wanted to share it with others. I am available to add or maintain this branch if you have any recommendations to make it better :)
Thanks for the work of @radu-matei, @dkim, @dllu, which is the base for this PR 🥇