分类：PyTorch模型转TensorRT

2021
09-07

PyTorch模型转TensorRT是怎么实现的?

转换步骤概览准备好模型定义文件（.py文件）准备好训练完成的权重文件（.pth或.pth.tar）安装onnx和onnxruntime将训练好的模型转换为.onnx格式安装tensorRT环境参数ubuntu-18.04PyTorch-1.8.1onnx-1.9.0onnxruntime-1.7.2cuda-11.1cudnn-8.2.0TensorRT-7.2.3.4PyTorch转ONNXStep1：安装ONNX和ONNXRUNTIME网上找到的安装方式是通过pippipinstallonnxpipinstallonnxruntime如果使用的是Anaconda环境，conda... 继续阅读 >