公众号关注 “ML_NLP”
设为 “星标”，重磅干货，第一时间送达！
大家好，我是Ray，今天给大家带来的是Pytorch的TensorRT转换器
Pytorch-> TensorRT
TensorRT是一个高性能的深度学习推理（Inference）优化器，可以为深度学习应用提供低延迟、高吞吐率的部署推理。TensorRT可用于对超大规模数据中心、嵌入式平台或自动驾驶平台进行推理加速。TensorRT现已能支持TensorFlow、Caffe、Mxnet、Pytorch等几乎所有的深度学习框架，将TensorRT和NVIDIA的GPU结合起来，能在几乎所有的框架中进行快速和高效的部署推理。
而今天给大家推荐的库能很方便的在Pytorch下使用TensorRT。
这个仓库可以通过pip install 下载
使用时只需调用一个函数就可以对模型进行转换
在各种模型下都有非常好的效果
仓库地址：
https://github.com/NVIDIA-AI-IOT/torch2trt
仓库地址共享：
在机器学习算法与自然语言处理公众号后台回复“代码”，
即可获取195篇NAACL+295篇ACL2019有代码开源的论文。开源地址如下：https://github.com/yizhen20133868/NLP-Conferences-Code
重磅！机器学习算法与自然语言处理交流群已正式成立！
群内有大量资源，欢迎大家进群学习！
额外赠送福利资源！邱锡鹏深度学习与神经网络，pytorch官方中文教程，利用Python进行数据分析，机器学习学习笔记，pandas官方文档中文版，effective java（中文版）等20项福利资源
获取方式：进入群后点开群公告即可领取下载链接
注意：请大家添加时修改备注为 [学校/公司 + 姓名 + 方向]
例如 —— 哈工大+张三+对话系统。
号主，微商请自觉绕道。谢谢！

推荐阅读：
开放域知识库问答研究回顾
使用PyTorch Lightning自动训练你的深度神经网络
PyTorch常用代码段合集

继续阅读
阅读原文
关键词
模型
深度学习
平台