onnxruntime::python::CreateExecutionProviderInstance CUDA_PATH is set but CUDA wasnt able to be load

错误提示:

RuntimeError:D:\a\_work\1\s\onnxruntime\python\onnxruntime_pybind_state.cc:891 onnxruntime::python::CreateExecutionProviderInstance CUDA_PATH is set but CUDA wasnt able to be loaded. Please install the correct version of CUDA andcuDNN as mentioned in the GPU requirements page  (https://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html#requirements),  make sure they're in the PATH, and that your GPU is supported.

1、根据这篇博客,确保cuda和cudnn的安装正确

win10下安装 Anaconda + Cuda + Cudnn + Pycharm + Pytorch_win10安装conda-CSDN博客

2、排除cuda是否正确安装问题,和确定cuda的版本

nvcc -V

3、版本cuda、cudnn和onnxruntime-gpu依赖关系,见官网Install ONNX Runtime | onnxruntime

cuda 版本为12.x时,请使用下面的方式进行重新安装

pip uninstall onnxruntime-gpu
pip install onnxruntime-gpu --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/

4、如果以上方法还不行,请执行安装命令

pip install onnxruntime-openvino

若出现拒绝访问Anaconda的环境时,根据提示可加参数 --user

pip install onnxruntime-openvino --user

5、最后个人开发环境在第4步,问题解决。

个人开发环境:cuda12.2 + cudnn8.9.2.26 + torch2.2.0

  • 3
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值