pip怎么安装onnx

阿杰姑娘_2244

阿杰姑娘_2244

2026-09-21

188人浏览

原创

安装onnx只需pip install onnx,它仅提供模型格式操作能力;真正运行模型需安装互斥的onnxruntime(cpu)或onnxruntime-gpu(gpu),且gpu版依赖匹配的cuda驱动与运行时库,并须显式启用cudaexecutionprovider。

pip怎么安装onnx

pip install onnx 就够了,这是安装 ONNX 格式工具包的标准命令。它不负责运行模型,只提供模型检查、修改、验证等能力(比如用 onnx.checker.check_model() 验证 .onnx 文件是否合法)。

但你真正想问的,大概率不是这个——而是“怎么装能跑 ONNX 模型的运行时”,也就是 onnxruntime。很多人搜“pip 安装 onnx”结果却卡在 import onnxruntime 报错,就是因为装错了包。


装 onnxruntime 之前必须确认硬件和 CUDA 版本

onnxruntimeonnxruntime-gpu 是两个互斥包,不能共存;装错一个,另一个就失效。

  • 没有 NVIDIA GPU,或只打算在 CPU 上跑:
    pip install onnxruntime

  • 有 NVIDIA GPU,且已装好 CUDA 驱动(注意:是驱动,不是 CUDA Toolkit):

    • CUDA 11.x(如 11.8):直接 pip install onnxruntime-gpu
    • CUDA 12.x(如 12.2 或 12.4):必须加额外源
      pip install onnxruntime-gpu --extra-index-url @#@#@#@#@#@#@#@#@#@0
  • 不确定 CUDA 版本?运行 nvidia-smi 查顶部显示的 “CUDA Version”(那是驱动支持的最高 CUDA 版本,不是你装的 Toolkit 版本)

    data-engineering-data-pipeline
    data-engineering-data-pipeline

    您是一位数据管道架构专家,专注于为批处理和流式数据处理构建可扩展、可靠且高性价比的数据管道。

    下载

PyCharm / 虚拟环境里 import 失败的常见原因

ModuleNotFoundError: No module named 'onnxruntime' 几乎从不因为包本身有问题,而是环境错位:

  • PyCharm 新建项目时没选对 Python 解释器,pip install 装到了系统 Python,但项目用的是虚拟环境
  • 终端里激活了 venv,但 PyCharm 的 Python 解释器设置仍指向全局 Python
  • 在 Conda 环境里混用 pip(尤其 conda installpip install 交叉使用),可能触发二进制不兼容
  • Windows 下用了 PowerShell 而没执行 venv\Scripts\Activate.ps1(默认被策略阻止),导致 pip 实际装到全局

验证是否装对了:在目标环境中运行
python -c "import onnxruntime; print(onnxruntime.<strong>version</strong>, onnxruntime.get_device())"
输出带 'GPU' 才说明 GPU 版生效了。


别用 pip 强装 onnxruntime-gpu 到无 GPU 机器

有人图省事,在笔记本上也硬装 onnxruntime-gpu,结果运行时报类似错误:
ONNXRuntimeError: [ONNXRuntimeErrorCode:1] : Load model failed: The specified module could not be found.

这不是模型问题,是 onnxruntime-gpu 依赖的 CUDA DLL(如 cublas64_11.dll)根本不存在。Windows 不报具体缺哪个文件,Linux 会提示 libcuda.so.1: cannot open shared object file

更隐蔽的坑是:某些云服务器(如 AWS g4dn)默认装了 NVIDIA 驱动但没装 CUDA Toolkit,onnxruntime-gpu 仍会加载失败——它需要的是驱动 + 对应版本的 CUDA 运行时库,二者缺一不可。


实际部署时最容易被忽略的一点:ONNX Runtime 的 GPU 支持不是“装了就自动加速”。你得显式启用 CudaExecutionProvider,否则即使 get_device() 返回 'GPU',推理也可能仍在 CPU 上走。初始化 session 时漏掉 providers=['CUDAExecutionProvider'],等于白装。

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

pip安装 pip pip命令

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
pip安装使用方法
pip安装使用方法

安装步骤:1、确保Python已经正确安装在您的计算机上;2、下载“get-pip.py”脚本;3、按下Win + R键,然后输入cmd并按下Enter键来打开命令行窗口;4、在命令行窗口中,使用cd命令切换到“get-pip.py”所在的目录;5、执行安装命令;6、验证安装结果即可。大家可以访问本专题下的文章,了解pip安装使用方法的更多内容。

2023.10.09

2350

5

更新pip版本
更新pip版本

更新pip版本方法有使用pip自身更新、使用操作系统自带的包管理工具、使用python包管理工具、手动安装最新版本。想了解更多相关的内容,请阅读专题下面的文章。

2024.12.20

5443

6

pip设置清华源
pip设置清华源

设置方法:1、打开终端或命令提示符窗口;2、运行“touch ~/.pip/pip.conf”命令创建一个名为pip的配置文件;3、打开pip.conf文件,然后添加“[global];index-url = https://pypi.tuna.tsinghua.edu.cn/simple”内容,这将把pip的镜像源设置为清华大学的镜像源;4、保存并关闭文件即可。

2024.12.23

6962

11

python升级pip
python升级pip

本专题整合了python升级pip相关教程,阅读下面的文章了解更多详细内容。

2025.07.23

3307

16

Vibeknow在线使用入口合集
Vibeknow在线使用入口合集

本专题汇总了Vibeknow在线创作视频的官方入口及网页版使用教程,涵盖PPT、PDF、Word等文档一键转讲解视频的核心操作,并整理了免费版水印规则与手机端浏览器访问指南,助你快速将知识内容视频化。

2026.09.21

0

20

NumPy随机数文件读写与dtype数据类型
NumPy随机数文件读写与dtype数据类型

本专题整理 NumPy 随机数、文件读写与 dtype 数据类型相关教程,覆盖 Generator/random、随机数种子、正态分布采样、npy/npz/CSV/TXT 保存读取、loadtxt/savetxt、memmap、大文件处理、astype 类型转换、结构化 dtype、整数溢出和精度丢失等场景。

2026.09.21

0

24

NumPy矩阵运算与线性代数计算
NumPy矩阵运算与线性代数计算

本专题整理 NumPy 矩阵运算与线性代数计算相关教程,覆盖矩阵乘法、dot 与 @ 运算符、逆矩阵、行列式、特征值与特征向量、SVD、线性方程组、欧氏距离、矩阵分解和大规模矩阵性能优化等内容,帮助读者掌握 np.linalg 与矩阵计算实战。

2026.09.21

0

20

NumPy广播机制数学运算与统计分析
NumPy广播机制数学运算与统计分析

本专题整理 NumPy 广播机制、数组数学运算与统计分析相关教程,覆盖广播规则、维度对齐、矩阵与数组加减除法、向量化计算、均值方差、分位数、中位数、直方图和 unique 频次统计等场景,帮助读者掌握 ndarray 高效计算与统计处理方法。

2026.09.21

0

17

NumPy数组创建索引切片与数据选择
NumPy数组创建索引切片与数据选择

本专题整理 NumPy 数组创建、索引、切片与数据选择相关教程,覆盖 np.array、zeros/ones、多维数组形状、基础切片、花式索引、布尔索引、条件筛选、视图与副本等常用场景,帮助读者系统掌握 ndarray 数据构造与高效提取方法。

2026.09.21

0

12

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
最新Python教程 从入门到精通
最新Python教程 从入门到精通

共4课时 | 22.8万人学习

麻省理工大佬Python课程
麻省理工大佬Python课程

共34课时 | 7.3万人学习

python基础快速入门
python基础快速入门

共18课时 | 3.3万人学习