windows - 新 LearningModelSession 中的 System.AccessViolationException（使用 GPU 模式）

Question

每个人！

我正在尝试基于 Windows-Machine-Learning github 的 SnapCandy 示例制作推理应用程序。

我的 ONNX 模型文件在 pytorch 1.10 中导出。Onnx 格式：v4，opset v9

Windows 版本：Win10 21H2 (Build# 19044.1466)

显卡：NDVIA 2060（驱动版本：8.1.961.0，VRAM 6GB）

代码：m_inferenceDeviceSelected = _useGPU？LearningModelDeviceKind.DirectXHighPerformance：LearningModelDeviceKind.Cpu；m_session = new LearningModelSession(m_model, new LearningModelDevice(m_inferenceDeviceSelected));

如果 _useGPU 为真（Gpu 模式），则发生 System.AccessViolationException。

但如果 _useGPU 为 false（Cpu 模式），则推理良好。

我该如何解决这个异常？

提前致谢！

Answer 1

我遇到了类似的问题，并在 github ( https://github.com/microsoft/Windows-Machine-Learning/issues/464 ) 上创建了一个问题，现在得到了回答：使用 DirectML 时，问题已解决。来自 \Windows\System32\ 的 DLL，而不是 nuget 数据包“Microsoft.AI.DirectML”附带的 DLL。最好的问候，奥利弗