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

Question

我不熟悉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)

Answer 1

看起来OrtValue是最近添加的（https://github.com/microsoft/onnxruntime/commit/6f54113a1bcea4e643de844ee33e4914145915ef），还没有发布

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