执行torch.onnx.export报错‘Module onnx is not installed!’求解决

1. 确认当前Python环境的onnx安装状态 #

先在运行代码的同一Python环境中执行以下代码，验证onnx是否真的安装成功：

import onnx
print(onnx.__version__)

如果执行报错，说明当前环境确实未安装onnx；如果能输出版本号，说明问题出在torch的检测逻辑或环境路径匹配上。

2. 检查Python环境一致性 #

• 若使用虚拟环境（venv/conda），确保激活目标环境后再执行安装命令，比如conda环境需先运行：

  conda activate your_env_name
  

• 验证pip与Python路径是否匹配，避免全局环境和虚拟环境混淆：

  # Linux/macOS
  which python
  which pip
  
  # Windows
  where python
  where pip
  

  若路径不匹配，改用python -m pip执行安装/卸载，确保操作针对当前环境：

  python -m pip uninstall -y torch torchvision onnx
  python -m pip install torch torchvision onnx
  

3. 安装适配PyTorch版本的onnx #

不同PyTorch版本对onnx有版本兼容要求，比如PyTorch 2.0.x建议安装onnx 1.13.x或1.14.x版本，指定版本安装尝试：

python -m pip install onnx==1.13.1

4. 清理PyTorch缓存 #

缓存异常可能导致检测失效，清理后重试：

• 删除PyTorch缓存目录：Linux/macOS为~/.cache/torch，Windows为C:\Users\你的用户名\.cache\torch
• 或在代码开头添加强制重新导入逻辑（临时修复）：

import importlib
import onnx
importlib.reload(onnx)
import torch
torch.onnx.export(...)

5. 验证IDE的Python解释器配置 #

如果在IDE（VSCode/PyCharm）中运行，检查IDE配置的解释器是否为已安装onnx的环境：

• PyCharm：进入File > Settings > Project: xxx > Python Interpreter确认路径
• VSCode：左下角切换Python解释器，选择对应环境

内容的提问来源于stack exchange，提问作者Paresh