问题发生在指定CUDAExecutionProvider的时候,不指定该项时(仅使用CPU)时不会出错。CUDA和CUDNN均正常安装。

解决方法:
安装正确匹配的onnxruntime-gpu和cuda、cudnn的版本。
来源:https://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

更多推荐