属性错误：模块“火炬”没有属性“inference_mode”

Question

我对

pytorch

非常陌生，当我尝试运行我的

CNN

时，我遇到了错误：

AttributeError：模块“torch”没有属性“inference_mode”。* 有谁知道发生了什么事吗？它在 Google colab 上有效，但在其他地方则无效。

Answer 1

尝试将版本升级到 1.13。参考：https://pytorch.org/docs/stable/ generated/torch.inference_mode.html

Answer 2

在较旧的 PyTorch 版本代码（可能低于 1.9.1）中，您可能必须使用 torch.no_grad() 代替 torch.inference_mode()：

示例：

with torch.inference_mode():  y_preds = model_0(X_test)

# 在旧版本中不起作用

with torch.no_grad(): y_preds = model_0(X_test)

将会完美工作

Answer 3

很可能是版本问题。

最近在

torch.inference_mode()

 中添加了 

v1.9

。确保您拥有正确的版本。

尝试打印

torch.__version__

来检查您的版本。