您当前的位置: 首页 > 

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

柳鲲鹏 发布时间:2019-04-16 16:14:14 ,浏览量:0

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)

 

关注
打赏
1688896170
查看更多评论

柳鲲鹏

暂无认证

  • 0浏览

    0关注

    4641博文

    0收益

  • 0浏览

    0点赞

    0打赏

    0留言

私信
关注
热门博文
立即登录/注册

微信扫码登录

0.0589s