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

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

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐