-
Notifications
You must be signed in to change notification settings - Fork 816
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
setup: Add version for torch dependency #1067
Conversation
Hard locks the package to a specific version of torch if defined Signed-off-by: Eli Uriegas <eliuriegas@fb.com>
Fix #1063 |
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.
Looks good!
Do we want to have this on |
|
||
pytorch_package_dep = 'torch' | ||
if pytorch_package_version is not None: | ||
pytorch_package_dep += "==" + pytorch_package_version |
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.
Does the cuda version etc factor in here?
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.
No it does not, but then again we don't account for cuda version in either audio
or vision
either
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. Does the package fail for the CI test?
Just fixed the CI tests. Let's re-run them before merging. |
I think CI failures are unrelated to this change, going to go ahead and merge |
setup: Add version for torch dependency
Hard locks the package to a specific version of torch if defined
Resolves #1063
Signed-off-by: Eli Uriegas eliuriegas@fb.com