|
报错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/
|
|