问题描述
File "/test/SDpytrirton/TensorRT-release-8.6/demo/Diffusion/utilities.py", line 213, in build network_from_onnx_path(onnx_path,flags=[trt.OnnxParserFlag.NATIVE_INSTANCENORM]),
TypeError: func_impl() got an unexpected keyword argument 'flags'
解决
-
pip install tensorrt==8.6.1 -i https://mirrors.aliyun.com/pypi/simple/
-
pip install polygraphy -i https://pypi.python.org/simple/ -
python3 -m pip install --upgrade polygraphy