python - 模块“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 或等到该功能正式发布。