登录 注册
当前位置:主页 > 资源下载 > 32 > 项目“朝圣者”提供的torch2trt工具,旨在将PyTorch模型高效转换为TensorRT引擎,适用于C/C++开发环境

项目“朝圣者”提供的torch2trt工具,旨在将PyTorch模型高效转换为TensorRT引擎,适用于C/C++开发环境

  • 更新:2024-12-11 23:39:32
  • 大小:345KB
  • 推荐:★★★★★
  • 来源:网友上传分享
  • 类别:其它 - 开发技术
  • 格式:ZIP

资源介绍

该项目由torch2trt和torch2trt_dynamic分叉,此项目的目的是提供一种将pytorch模型直接转换为TensorRT引擎的朝圣者项目。 TensorRT引擎。 如果您的模型是基于pytorch构建的,则可以加快您的项目的工作速度;我们已经测试了多个模型,这些模型都可以正常工作,并且我们将添加更多测试和插件来支持复杂模型的模型。 我们不希望使用onnx的原因是:onnx是另一种不太必要的中间件; 维护onnx插件并不容易