Skip to content

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

Open
CorentinJ wants to merge 2 commits intoELS-RD:mainfrom
resemble-ai:main
Open

Replaced tensorrt.tensorrt imports to tensorrt for pip package compatibility#181
CorentinJ wants to merge 2 commits intoELS-RD:mainfrom
resemble-ai:main

Conversation

@CorentinJ
Copy link
Copy Markdown

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

Development

Successfully merging this pull request may close these issues.

1 participant