![](https://img-blog.csdnimg.cn/20201014180756738.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
TensorRT
文章平均质量分 54
zengyiii
这个作者很懒,什么都没留下…
展开
-
PyTorch 预训练模型转 onnx,Tensorrt 部署
PyTorch 预训练模型转 onnx,Tensorrt 部署所用环境Centos 7PyTorch 1.8.0(此为模型训练时的配置,转 onnx 时改为1.2.0)onnx 1.7.0tensorrt 7.2.2.3cuda 11.1安装 onnxpip install onnx安装 tensorrt另一位博主 @幸福回头 的博文中 (https://blog.csdn.net/zt1091574181/article/details/100972562) 对于安装 tenso原创 2021-07-28 15:25:16 · 1025 阅读 · 4 评论 -
onnx与tensorrt转换:Assertion failed: dims.nbDims == 4 || dims.nbDims == 5
使用TensorRT-7.2.2.3转换onnx为trt文件时出现错误:Assertion failed: dims.nbDims == 4 || dims.nbDims == 5[8] Assertion failed: dims.nbDims == 4 || dims.nbDims == 5[07/28/2021-12:10:20] [E] Failed to parse onnx file[07/28/2021-12:10:20] [E] Parsing model failed[07/28/原创 2021-07-28 15:34:57 · 654 阅读 · 0 评论