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,报空指针错误,不知道有人遇到过没有?
[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
你好,我是有问必答小助手,非常抱歉,本次您提出的有问必答问题,技术专家团超时未为您做出解答
本次提问扣除的有问必答次数,将会以问答VIP体验卡(1次有问必答机会、商城购买实体图书享受95折优惠)的形式为您补发到账户。
因为有问必答VIP体验卡有效期仅有1天,您在需要使用的时候【私信】联系我,我会为您补发。