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

ImportError: cannot import name 'create_trt_engine' from 'mmdeploy.backend.tensorrt' #1019

Closed
Oscillated opened this issue Sep 7, 2022 · 7 comments · Fixed by #1021
Closed
Assignees

Comments

@Oscillated
Copy link

I follow the tutorial, and build SDK without errors. But find errors when I try to add plugins into TensorRT backends. (follow 第七章: TENSORRT 自定义插件 )

image

This is my env

image

image

@tpoisonooo
Copy link
Collaborator

Have you python3 -m pip install /path/to/tensorrt/python/tensorrt-pyxxxx-.whl ?

@irexyc
Copy link
Collaborator

irexyc commented Sep 7, 2022

you also shoud install pycuda through pip install pycuda

@Oscillated
Copy link
Author

Oscillated commented Sep 7, 2022

Have you python3 -m pip install /path/to/tensorrt/python/tensorrt-pyxxxx-.whl ?

Yes, I have installed tensorrt-8.4.1.5-cp38-none-linux_x86_64.whl in tensorrt directory.

@Oscillated
Copy link
Author

you also shoud install pycuda through pip install pycuda

Thanks for advice, but I have also installed pycuda during the setup process.

@AllentDan
Copy link
Member

Hi, @Oscillated . create_trt_engine has been deprecated in the latest MMDeploy. Please use from_onnx instead.

@Oscillated
Copy link
Author

Hi, @Oscillated . create_trt_engine has been deprecated in the latest MMDeploy. Please use from_onnx instead.

I re-build mmdeploy, now it is version 0.8.0, but it still seems to have errors when I follow the example of from_onnx function. How can I solve it ? Thanks.

error

@AllentDan
Copy link
Member

You may refer to PR #1021.

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 a pull request may close this issue.

5 participants