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

Add a hook for torchvision.ops #80

Merged
merged 4 commits into from
Dec 15, 2020
Merged

Conversation

rokm
Copy link
Member

@rokm rokm commented Dec 15, 2020

Add a hidden import of torchvision._C extension module that we fail to pick up automatically. This extension module is required in functions from torchvision.ops.* modules, where its absence causes the _assert_has_ops() call to fail.

Fixes pyinstaller/pyinstaller#5386

Add a hidden import of torchvision._C extension module that we
fail to pick up automatically. This extension module is required
in functions from torchvision.ops.* modules, where its absence
causes the _assert_has_ops() call to fail.

Fixes the failing test_torchvision_nms.
Fixes pyinstaller/pyinstaller#5386.
@rokm rokm requested review from a team and bwoodsend and removed request for a team December 15, 2020 13:40
@bwoodsend
Copy link
Member

Can you run the CI run tests? Maybe just do one version of Python?

@rokm
Copy link
Member Author

rokm commented Dec 15, 2020

Here's the run for the latest stable version under python 3.9 on Windows and Linux: https://github.com/rokm/pyinstaller-hooks-contrib/runs/1557959180?check_suite_focus=true.

@bwoodsend bwoodsend merged commit 861fbf6 into pyinstaller:master Dec 15, 2020
@rokm rokm deleted the hooks-torchvision branch September 26, 2021 08:37
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

can't automatic add torchvision to my .exe package
2 participants