Skip to content
New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # 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

rokm added 3 commits December 15, 2020 12:25
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
# for free to join this conversation on GitHub. Already have an account? # 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