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/

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