본문 바로가기

카테고리 없음

Pyinstaller시 MMDeploy NotImplementedError: Unsupported backend type: tensorrt 오류

분명 python3로 실행하면 오류가 안나는데, pyinstaller로 빌드하면 저런 오류가 난다.

 

최대한 해결해보려고 했으나, 감을 못잡았고..

 

File "mmdeploy/codebase/base/backend_model.py", line 63, in _build_wrapper

 

여기에서 난다는 것을 보았다.

 

그래서 해당 backend_model.py에 들어갔으며,

 

항상 None으로 출력되는 줄 위에 import를 직접 시켜줘서 넣어줘야겠다고 생각했음.

 

하드하긴하지만 일단 해결은 봐야했기 때문에

 

기존 코드는 아마 _build_wrapper함수안에 이렇게 되어있을 것

from mmdeploy.backend.base import get_backend_manager

backend_mgr = get_backend_manager(backend.value)

 

근데 backend_mgr이 항상 None으로 나옴

원래 trt 클래스가 담겨야함.

 

그래서 직접 넣어주었음

 

from mmdeploy.backend.tensorrt.backend_manager import TensorRTManager

backend_mgr = TensorRTManager

 

이렇게 넣어주니 성공...

 

미리 import를 해주면 그제서야 인식하는 경우들이 있었는데, 이번에도 동일했음.

None으로 나오던게, 저 줄 하나로 다시 TRT Manager를 인식했음...

 

무튼 해결