在安装onnxruntime时,如果直接运行“pip install onnxruntime-gpu”安装的是最新版的onnxruntime-gpu,且对应的是的(CUDA 12.x)。这样很有可能导致CUDA版本不对应导致GPU用不了。
CUP版安装
https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT-Nightly/pypi/simple/ onnxruntime
GPU版安装(CUDA 12.x)
对应的是onnxruntime-gpu-1.19.2
pip install --pre --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT-Nightly/pypi/simple/ onnxruntime-gpu
GPU版安装(CUDA 11.x)
对应的是onnxruntime-gpu-1.19.2
pip install onnxruntime-gpu --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-11/pypi/simple/
参考官网链接:https://onnxruntime.ai/docs/install/
2790

被折叠的 条评论
为什么被折叠?



