报错:
[E:onnxruntime:Default, provider_bridge_ort.cc:1745 onnxruntime::TryGetProviderInfo_CUDA] D:\a\_work\1\s\onnxruntime\core\session\provider_bridge_ort.cc:1426 onnxruntime::ProviderLibrary::Get [ONNXRuntimeError] : 1 : FAIL : LoadLibrary failed with error 126 "" when trying to load "F:\ProgramData\anaconda3\envs\py38gpu\lib\site-packages\onnxruntime\capi\onnxruntime_providers_cuda.dll"
EP Error 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.
when using [('CUDAExecutionProvider', {'device_id': 0, 'arena_extend_strategy': 'kNextPowerOfTwo', 'gpu_mem_limit': 2147483648, 'cudnn_conv_algo_search': 'DEFAULT'}), 'CPUExecutionProvider']
Falling back to ['CUDAExecutionProvider', 'CPUExecutionProvider'] and retrying.
onnx与cuda+cudnn版本:
这个原因如果cuda和cudnn安装的没问题的话,版本对应onnx查找这个网站:NVIDIA - CUDA | onnxruntime
排除掉cuda和cudnn的版本问题,我的是cuda12.1+cudnn9.1,应该安装onnx1.18.1
解决:
使用默认pip install onnxruntime-gpu==1.18.1的话,安装的版本不一定对应,应指定cuda12的版本,pip信息如下:
pip install onnxruntime-gpu==1.18.1 --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/
把1.18.1可以换成你自己的版本,最后链接onnxruntime-cuda-12也换成你的cuda版本
终于解决这个问题,mark一下。
参考链接:【已解决】onnx无法找到CUDA的路径_onnxruntime-gpu 1.18.1 cudnn 9.x.-CSDN博客
ComfyUI: 报EP Error错误(onnxruntime)_falling back to ['cudaexecutionprovider', 'cpuexec-CSDN博客