tensorrt构建engine是老是报空指针错误

Question

def build_engine(model_file):
#     TRT_LOGGER = trt.Logger(trt.Logger.WARNING)
    with trt.Builder(TRT_LOGGER) as builder, builder.create_network() as network,trt.OnnxParser(network, TRT_LOGGER) as parser:
        builder.max_workspace_size = MAX_WORKSPACE_SIZE
        builder.max_batch_size = MAX_BATCH_SIZE
        with open(model_file, 'rb') as model:
            parser.parse(model.read())
            engine = builder.build_cuda_engine(network)
    return engine

就代码中的with trt.Builder(TRT_LOGGER) as builder,报空指针错误，不知道有人遇到过没有？

Answer 1

[679] Traceback (most recent call last):
[680]   File "/xiaoou/STDC-Seg-master/STDC-Seg-master/latentcy/run_latency_stages.py", line 98, in <module>
[681]     main() 
[682]   File "/xiaoou/STDC-Seg-master/STDC-Seg-master/latentcy/run_latency_stages.py", line 84, in main
[683]     latency = compute_latency(model, inputDimension)
[684]   File "/xiaoou/STDC-Seg-master/STDC-Seg-master/latentcy/utils/darts_utils.py", line 173, in compute_latency_ms_tensorrt
[685]     with build_engine("./model.onnx") as engine:
[686]   File "/xiaoou/STDC-Seg-master/STDC-Seg-master/latentcy/utils/darts_utils.py", line 120, in build_engine
[687]     with trt.Builder(TRT_LOGGER) as builder, builder.create_network() as network, trt.OnnxParser(network, TRT_LOGGER) as parser:
[688] TypeError: pybind11::init(): factory function returned nullptr