天天看點

TensorRT:AttributeError: 'module' object has no attribute 'Logger'

TensorRT版本與代碼寫法不對.

具體寫法:

# using trt 4.0
        G_LOGGER = trt.infer.ConsoleLogger(trt.infer.LogSeverity.INFO)
        trt_engine = trt.utils.load_engine(G_LOGGER, engine_path)
 
        """
        ## using trt 5.0
        G_LOGGER = trt.Logger(trt.Logger.WARNING)
        with open(engine_path, "rb") as f, trt.Runtime(G_LOGGER) as runtime:
            trt_engine = runtime.deserialize_cuda_engine(f.read())
        """
 
        self.engine = Engine(trt_engine)      

繼續閱讀