导出 .onnx 文件时未分配提供商、rl-agents 的问题

Question

我试图使用 rl-agents 和 godot 创建一个 AI。一切都很顺利，直到导出过程。

出现以下错误：

提高值错误（ ValueError：此 ORT 版本已启用 ['AzureExecutionProvider', 'CPUExecutionProvider']。从ORT 1.9开始，实例化InferenceSession时需要显式设置providers参数。例如，onnxruntime.InferenceSession(...,providers=['AzureExecutionProvider','CPUExecutionProvider'],...)

我尝试在 onnxruntime.InferenceSession() 函数中包含它告诉我的提供者，但没有任何改变。

应该是错误的文件是：stable_baselines3.py、stable_baselines_export.py、onnxruntime_inference_collection.py。

我对这一切都很陌生，只是想解决它。

尝试导出 .onnx 文件，但导出时出错。那么该文件就无法使用了。

Answer 1

我通过显式定义这个参数解决了这个问题：

providers=['CUDAExecutionProvider', 'CPUExecutionProvider']