问题发生在指定CUDAExecutionProvider的时候,不指定该项时(仅使用CPU)时不会出错。CUDA和CUDNN均正常安装。
解决方法:
安装正确匹配的onnxruntime-gpu和cuda、cudnn的版本。
来源:https://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html
问题发生在指定CUDAExecutionProvider的时候,不指定该项时(仅使用CPU)时不会出错。CUDA和CUDNN均正常安装。
解决方法:
安装正确匹配的onnxruntime-gpu和cuda、cudnn的版本。
来源:https://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html