ModuleNotFoundError：没有名为“onnxruntime”的模块

Question

我正在学习 Microsoft PyTorch 课程并尝试在 Kaggle Notebooks 上实现，但我一遍又一遍地收到相同的错误消息：“ModuleNotFoundError：没有名为 'onnxruntime' 的模块”。我已经检查了所有可能的地方是否可以找到解决方案，但没有，我什至尝试在笔记本中使用 pip 手动安装它，但它仍然无法正常工作。我检查了官方 onnxruntime 网站和文档，但没有任何地方说明任何有关过时或任何其他问题的内容。有人帮忙。我的代码无法运行，因为它显示“onnxruntime 未定义”。这是我的进口：

    %matplotlib inline
import torch
import onnxruntime
from torch import nn
import torch.onnx as onnx
import torchvision.models as models
from torchvision import datasets
from torchvision.transforms import ToTensor

以及我正在尝试运行的代码单元

session = onnxruntime.InferenceSession(onnx_model, None)
input_name = session.get_inputs()[0].name
output_name = session.get_outputs()[0].name

result = session.run([output_name], {input_name: x.numpy()})
predicted, actual = classes[result[0][0].argmax(0)], classes[y]
print(f'Predicted: "{predicted}", Actual: "{actual}"')

您可以在这里找到完整的笔记本：https://www.kaggle.com/faisalalbasu/complete-model

Answer 1

• 发生错误是因为“import”在任何路径中都找不到onnxruntime，请检查import正在搜索的位置并查看onnxruntime是否在那里。
• 检查 pip install 安装到哪个路径，这样以后就不会遇到同样的问题了！ :)

Answer 2

如何检查 import 在终端中搜索的位置，以及如何检查 pip install 安装到的路径....也会有所帮助。 我检查了 onnxruntime，并在 python 文件夹中按预期找到了它，但是仍然是相同的旧答案： 文件“/Users/name/roop-main/run.py”，第 3 行，位于 从roop导入核心 文件“/Users/name/roop-main/roop/core.py”，第 16 行，位于 导入nx运行时 ModuleNotFoundError：没有名为“onnxruntime”的模块