tensorrt踩坑日记(python pytorch 转 onnx 推理加速)
配置:Ubuntu 16.04python 3.6onnx 1.6pytorch 1.5pycuda 2019.1.2torchvision 0.1.8建议详读,先安装好环境官网指导手册步骤:1.将pytorch模型转换成onnx模型这边用的是Darknet生成的pytoch模型import torchfrom torch.autograd import Variableimport onnxinput_name = ['input']output_n
原创
2020-08-01 16:59:47 ·
4031 阅读 ·
2 评论