Enable JavaScript to run this website.
首页
标签
分类
归档
友链
关于
标签: PyTorch
共1篇文章
2023年7月10日
共10995字
约8分钟
TensorRT踩坑教程
深度学习/模型部署
PyTorch
TensorRT
CUDA
0. 前言 在部署深度学习模型时需要用到TensorRT,看了一遍文档后,我发现对于初学者来说,官方文档实在是太晦涩。文档里既没有接口的应用示例,也没有及时更新新接口的用法。在翻阅了部分博客和TensorRT仓库源码后,才有了本文。 TensorRT是NVIDIA推出的用于深度学习推理的高性能引擎,旨在优化深度神经网络模型的推理速度和性能。TensorRT可以通过层融合、混合精度、量化等技术显著提高深度神经网络的推理性能。 NVIDI…