模块“ onnxruntime”没有属性“ OrtValue”

问题描述

我是python和onnxruntime的新手。我已经成功地进行了推理。我目前正在尝试通过IObinding将推理实例定向到系统上的特定gpu。尝试创建OrtValue时收到以下错误：AttributeError：模块'onnxruntime'没有属性'OrtValue'。到目前为止，我还没有找到解决方案。希望有人能对此事有所启发。这是一些信息。

环境

OS: Windows 10

Python version: 3.7.8

OnnxRuntime-gpu: 1.5.2 install via pip

Cuda version: 10.2

CudNN vesion: 8.03.33

错误代码：

X_ortvalue = ort.OrtValue.ortvalue_from_numpy(ort_inputs,'cuda',0)

解决方法

似乎OrtValue是最近添加的内容（https://github.com/microsoft/onnxruntime/commit/6f54113a1bcea4e643de844ee33e4914145915ef），并且尚未发布。

您可以从源代码构建onnxruntime，也可以等到该功能正式发布。