您当前的位置: 首页 >  pytorch

Pytorch/Caffe可以先转换为ONNX,再转换为TensorRT

柳鲲鹏 发布时间:2018-11-01 12:53:30 ,浏览量:0

  近来工作,试图把Pytorch用TensorRT运行。折腾了半天,没有完成。github中的转换代码,只能处理pytorch 0.2.0的功能(也明确表示不维护了)。和同事一起处理了很多例外,还是没有通过。吾以为,实际上即使勉强过了,能不能跑也是问题。

  后来有高手建议,先转换为ONNX,再转换为TensorRT。这个思路基本可行。

 

  是不是这样就万事大吉?当然不是,还是有严重问题要解决的。这只是个思路。

  

关注
打赏
1688896170
查看更多评论

柳鲲鹏

暂无认证

  • 0浏览

    0关注

    4641博文

    0收益

  • 0浏览

    0点赞

    0打赏

    0留言

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

微信扫码登录

0.0476s