ONNX安装

前言

以下均用清华镜像源安装,官网太慢。
CPU和GPU版,只能安装一个,建议安装 GPU 版。
安装时一定要注意与CUDA、cuDNN版本适配问题,具体适配列表参考:CUDA Execution Provider

安装 onnx

pip install onnx -i https://pypi.tuna.tsinghua.edu.cn/simple

安装 onnxruntime(CPU版)

pip install onnxruntime -i https://pypi.tuna.tsinghua.edu.cn/simple

安装 onnxruntime(GPU版)

pip install onnxruntime-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple

常用国内镜像源

  • 清华大学:https://pypi.tuna.tsinghua.edu.cn/simple
  • 中国科学技术大学 : https://pypi.mirrors.ustc.edu.cn/simple
  • 豆瓣:http://pypi.douban.com/simple/
  • 阿里云:http://mirrors.aliyun.com/pypi/simple/

猜你喜欢

转载自blog.csdn.net/gdxb666/article/details/127575947