pytorch - AttributeError：模块“火炬”没有属性“推理模式”

Question

我对 pytorch 很陌生，当我尝试运行我的 CNN 时，我遇到了错误： AttributeError：模块 'torch' 没有属性 'inference_mode'。 有谁知道发生了什么？它在谷歌 colab 上工作，但在其他地方没有。

标签： pytorch

Answer 1

很可能是版本问题。

torch.inference_mode()最近在v1.9. 确保您拥有正确的版本。

尝试打印torch.__version__以检查您的版本。