1. 程式人生 > >Pytorch/Caffe可以先轉換為ONNX,再轉換為TensorRT

Pytorch/Caffe可以先轉換為ONNX,再轉換為TensorRT

  近來工作,試圖把Pytorch用TensorRT執行。折騰了半天,沒有完成。github中的轉換程式碼,只能處理pytorch 0.2.0的功能(也明確表示不維護了)。和同事一起處理了很多例外,還是沒有通過。吾以為,實際上即使勉強過了,能不能跑也是問題。

  後來有高手建議,先轉換為ONNX,再轉換為TensorRT。這個思路基本可行。

  是不是這樣就萬事大吉?當然不是,還是有嚴重問題要解決的。這只是個思路。