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

Replaced tensorrt.tensorrt imports to tensorrt for pip package compatibility #181

Open
wants to merge 2 commits into
base: main
Choose a base branch
from

Conversation

CorentinJ
Copy link

The tensorrt pip package seems not to have tensorrt.tensorrt as module (at least as of version tensorrt==8.6.1.post1), causing 3 imports to fail in the repo. The same members imported from tensorrt.tensorrt are present directly in tensorrt.
It seems that a manual install of Tensor RT (through an archive or .deb package) does not have this problem, but that the members present in tensorrt.tensorrt are equally present in tensorrt.

I haven't gone through the tensorrt changelog to figure out what happened there; but unless this breaks compatibility with older tensorrt versions, changing all tensorrt.tensorrt imports to tensorrt will work with both install methods.

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.

1 participant