-
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: install pytorch cu116 for server docker image #882
Conversation
0ae32d6
to
77ac532
Compare
please rebuild the image and reply to the communities. |
Codecov Report
@@ Coverage Diff @@
## main #882 +/- ##
===========================================
- Coverage 83.06% 71.78% -11.28%
===========================================
Files 22 22
Lines 1529 1531 +2
===========================================
- Hits 1270 1099 -171
- Misses 259 432 +173
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. |
@ZiniuYu Does the hub executor also suffer from this issue? |
I just tested both Torch and ONNX executors, they are all fine |
Then, I doubled if this issue comes from cuda? And I would not recommend pinning the cuda version. Our work should work on all stable cuda versions. |
So, please try to make it work with |
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
This PR fixes the error in linked issue