定位和解决 pip install onnxruntime-gpu
报错,无法安装的问题
报错信息如下:
ERROR: Could not find a version that satisfies the requirement onnxruntime-gpu (from versions: none)
ERROR: No matching distribution found for onnxruntime-gpu
先说结论:
原因是 python 版本太高了,重新创建一个稍低版本(python 3.10)的虚拟环境即可, 如下。
conda create --name alan_ai python=3.10
# 再执行:
pip install onnxruntime-gpu --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/
我之前是 conda create --name alan_ai python=3
安装的,默认给我装的 3.12.14 的版本,当前时间节点下,onnxruntime-gpu 没有相应版本的包。
服务器环境如下:
python.version: 3.12.4
torch.version: 2.4.0+cu121
CUDA.version: 12.1
cuDNN.version: 90100
尝试过:
pip install onnxruntime-gpu --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/
pip install onnxruntime-gpu==1.18.1 --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/
以下源地址都试过,都不行:
https://pypi.tuna.tsinghua.edu.cn/simple
https://pypi.ngc.nvidia.com
https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/
相关链接:
https://onnxruntime.ai/docs/install/
https://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html
https://pypi.org/project/onnxruntime-gpu/
https://aiinfra.visualstudio.com/PublicPackages/_artifacts/feed/onnxruntime-cuda-12/PyPI/onnxruntime-gpu/overview/1.18.1
https://pypi.org/project/onnxruntime/#files
https://github.com/KwaiVGI/LivePortrait/issues/240