-
Notifications
You must be signed in to change notification settings - Fork 2.1k
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
fix: onnx package conflict during setup #894
Conversation
6e07f2e
to
b9dcde3
Compare
Codecov Report
@@ Coverage Diff @@
## main #894 +/- ##
===========================================
- Coverage 83.08% 71.60% -11.49%
===========================================
Files 22 23 +1
Lines 1531 1567 +36
===========================================
- Hits 1272 1122 -150
- Misses 259 445 +186
Flags with carried forward coverage won't be shown. Click here to find out more.
Help us with your feedback. Take ten seconds to tell us how you rate us. Have a feature suggestion? Share it here. |
server/setup.py
Outdated
'onnx', | ||
'onnxmltools', | ||
'onnxruntime', |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
As you mentioned, we can always install onnxruntime-gpu
?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I've tested on x86 machines with/without GPU, both work fine. The CI failed for unknown reasons
📝 Docs are deployed on https://ft-fix-onnx-docker--jina-docs.netlify.app 🎉 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
There is a package conflict with
onnxruntime
andonnxruntime-gpu
in setup. If we install both packages, we are not able to run onnx using gpu. Installing onlyonnxruntime-gpu
enables clip_server to run both on CPU and GPU.This PR also corrects the instructions on running docker image in different runtimes.
This PR also replaces the broken images in test files
Notes: onnxruntime is fixed at 1.13.1 to pass the test_ranker
nvidia-tensorrt is fixed at 8.4.1.5 to pass trt related tests