找回密码
 会员注册
查看: 37|回复: 0

已解决onnx无法找到CUDA的路径

[复制链接]

4

主题

0

回帖

13

积分

新手上路

积分
13
发表于 2024-9-11 20:11:40 | 显示全部楼层 |阅读模式
报错RuntimeError:\a\_work\1\s\onnxruntime\python\onnxruntime_pybind_state.cc:857onnxruntime::python::CreateExecutionProviderInstanceCUDA_PATHissetbutCUDAwasntabletobeloaded.PleaseinstallthecorrectversionofCUDAandcuDNNasmentionedintheGPUrequirementspage (https://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html#requirements), makesurethey'reinthePATH,andthatyourGPUissupported. 问题分析onnx找到了gpu,但是还是不能调用解决办法先检查一下cudnn的环境变量配置是否正确,如果不正确,先看看怎么安装cudnn接下来是环境变量没问题的情况        环境变量没问题的话,一般就是cudnn和cuda以及onnx的版本不匹配        版本匹配看这里        查看好自己cudnn和cuda对应onnx版本后,输入这个pipinstallonnxruntime-gpu==what_you_want_version--extra-index-urlhttps://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/        将what_you_want_version换成你需要的版本,如我是cuda12.x以及cudnn8.x        那么就是pipinstallonnxruntime-gpu==1.18.1--extra-index-urlhttps://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 会员注册

本版积分规则

QQ|手机版|心飞设计-版权所有:微度网络信息技术服务中心 ( 鲁ICP备17032091号-12 )|网站地图

GMT+8, 2024-12-27 02:59 , Processed in 0.374419 second(s), 25 queries .

Powered by Discuz! X3.5

© 2001-2024 Discuz! Team.

快速回复 返回顶部 返回列表