(报错解决)[E:onnxruntime:Default, provider_bridge_ort.cc:1862 TryGetProviderInfo_CUDA]

报错内容:
在使用onnxruntime进行onnx模型文件推理时遇到如下报错:[E:onnxruntime:Default, provider_bridge_ort.cc:1862 TryGetProviderInfo_CUDA] /onnxruntime_src/onnxruntime/core/session/provider_bridge_ort.cc:1539 onnxruntime::Provider& onnxruntime::ProviderLibrary::Get() [ONNXRuntimeError]
以上报错会导致不能使用GPU推理而使用CPU执行推理。

解决办法:
原因是onnxruntime-gpu版本和当前设备CUDA和Cudnn版本不匹配导致。
https://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html前往onnxruntime官网查询匹配版本,安装指定版本即可。
我是将onnxruntime-gpu==1.20.1降级到1.18.0即可使用GPU推理。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值