Runtime
-
說明:
capybara.runtime提供Runtime與Backend的註冊表,用來描述「推論 runtime」與「實際執行後端(provider/device)」的對應關係。 -
常見 Runtime
Runtime.onnx:ONNXRuntimeRuntime.openvino:OpenVINORuntime.pt:TorchScript(PyTorch)
-
常用操作
Runtime.<name>.available_backends():取得該 runtime 支援的後端列表(依定義順序)。Runtime.<name>.normalize_backend(backend):把None/str/Backend正規化成Backend(None會使用 default backend)。Runtime.<name>.auto_backend_name():依環境自動選擇後端(例如優先使用 CUDA)。
-
注意
Backend.from_any(value, runtime=...)在多 runtime 共存時需要指定runtime,否則會拋出ValueError(以目前實作為準)。
-
範例
from capybara.runtime import Runtime
runtime = Runtime.onnx
print([b.name for b in runtime.available_backends()])
print(runtime.auto_backend_name())