Skip to content

Latest commit

 

History

History
47 lines (36 loc) · 1.22 KB

how_to_change_backend.md

File metadata and controls

47 lines (36 loc) · 1.22 KB

如何切换模型推理后端

FastDeploy中各视觉模型可支持多种后端,包括

  • OpenVINO (支持Paddle/ONNX两种格式模型, 仅支持CPU上推理)
  • ONNX Runtime (支持Paddle/ONNX两种格式模型, 支持CPU/GPU)
  • TensorRT (支持Paddle/ONNX两种格式模型,仅支持GPU上推理)
  • Paddle Inference(支持Paddle格式模型, 支持CPU/GPU)

所有模型切换后端方式均通过RuntimeOption进行切换,

Python

import fastdeploy as fd
option = fd.RuntimeOption()

# 切换使用CPU/GPU
option.use_cpu()
option.use_gpu()

# 切换不同后端
option.use_paddle_backend() # Paddle Inference
option.use_trt_backend() # TensorRT
option.use_openvino_backend() # OpenVINO
option.use_ort_backend() # ONNX Runtime

C++

fastdeploy::RuntimeOption option;

// 切换使用CPU/GPU
option.UseCpu();
option.UseGpu();

// 切换不同后端
option.UsePaddleBackend(); // Paddle Inference
option.UseTrtBackend(); // TensorRT
option.UseOpenVINOBackend(); // OpenVINO
option.UseOrtBackend(); // ONNX Runtime

具体示例可参阅FastDeploy/examples/vision下不同模型的python或c++推理代码

更多RuntimeOption的配置方式查阅FastDeploy API文档