溫馨提示×

c++ tensorrt支持哪些框架

c++
小樊
82
2024-07-24 14:13:13
欄目: 編程語言

TensorRT支持的深度學習框架包括但不限于：

  1. TensorFlow
  2. PyTorch
  3. ONNX
  4. Caffe
  5. MXNet

TensorRT還支持自定義網(wǎng)絡結構，允許用戶將自己實現(xiàn)的模型集成到TensorRT中進行加速推斷。TensorRT還提供了一系列的API和工具，方便用戶對模型進行優(yōu)化和部署。

0 踩