运行时错误：在您的系统上找不到 NVIDIA 驱动程序

Posted 2023-03-29

【中文标题】运行时错误：在您的系统上找不到 NVIDIA 驱动程序

【英文标题】：Runtime Error: Found no NVIDIA driver on your system

【发布时间】：2021-08-16 15:46:45

【问题描述】：

我在使用使用 GPU 训练的 Torch 加载模型时遇到问题，我正在尝试使用 CPU 加载该模型。但是我能够成功加载模型，但是在预测结果时出现错误。但是，如果我使用 GPU 机器，我可以预测输出，但不能预测 CPU：

我的代码：

****To save the model I am using :****
PATH = "model.pt"
torch.save(model, PATH)

**To Load the Model**


 import torch
    PATH = "model.pt"
    device = torch.device('cpu')
    loaded_model=torch.load(PATH, map_location=device)

我能够成功加载模型。但是在预测我遇到运行时错误时

**Predicting the loaded model using CPU**
predicted_title = loaded_model.predict([abstract])

。请检查您是否有 NVIDIA GPU 并安装了驱动程序

如果错误可能非常简单，但我无法纠正，我很抱歉。

【问题讨论】：

根据documentation 这应该可以工作。您能否验证摘要是否也在 CPU 设备上？

【参考方案1】：

你可以输出模型设备

print(loaded_model.device)

如果不是cpu，做

model = model.to('cpu')