AttributeError:模块“火炬”没有属性“inference_mode”

Question

当我尝试运行我的CNN时，我遇到了一个错误: AttributeError:模块'torch‘没有属性’推论_模式‘。有人知道怎么回事吗？它在Google上工作，但在其他地方却没有用。

Answer 1

很可能是版本问题。

torch.inference_mode()是最近在v1.9中添加的。确保你有正确的版本。

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