pip安装onnxruntime_gpu1.6.0失败

pip安装onnxruntime_gpu1.6.0失败

安装onnxruntime_gpu时需要对应cuda版本安装,笔者安装的cuda版本为10.2,对应onnxruntime_gpu应该安装1.5-1.6的,但是直接用命令pip install onnxruntime-gpu 安装的版本是最新版,指定版本安装pip install onnxruntime-gpu==1.6.0同样报错

Looking in indexes: https://pypi.tuna.tsinghua.edu.cn/simple, https://pypi.ngc.nvidia.com
ERROR: Could not find a version that satisfies the requirement onnxruntime-gpu==1.5.2 (from versions: 1.7.0, 1.8.0, 1.8.1, 1.9.0, 1.10.0, 1.11.0, 1.11.1)
ERROR: No matching distribution found for onnxruntime-gpu==1.5.2

找不到指定版本。
折腾半天找到方法:
在https://pypi.org/project/onnxruntime-gpu/1.6.0/#files找到需要的版本,其他版本可点击pip安装onnxruntime_gpu1.6.0失败_第1张图片
选择版本号,再点击download
pip安装onnxruntime_gpu1.6.0失败_第2张图片
根据需求下载轮子
pip安装onnxruntime_gpu1.6.0失败_第3张图片

下载好后激活需要安装的环境。
这里笔者使用pycharm
直接pip install xx.whl报错了
在这里插入图片描述
需要用命令pip debug --verbose查看轮子命名规则
pip安装onnxruntime_gpu1.6.0失败_第4张图片
这里我将轮子名称改为onnxruntime_gpu-1.6.0-cp37-abi3-win_amd64.whl
然后pip install
pip安装onnxruntime_gpu1.6.0失败_第5张图片
conda list查看
在这里插入图片描述
安装成功!

你可能感兴趣的:(深度学习,python,pycharm,深度学习)