AttributeError: module ‘torch‘ has no attribute ‘inference_mode‘

在运行论文源码时弹出AttributeError: module 'torch' has no attribute 'inference_mode' 异常

原因是pytorch版本问题

torch.inference_mode()在v1.9发布的,所以要保证pytorch版本在1.9以上。

pytorch 以前版本安装命令网站

https://pytorch.org/get-started/previous-versions/

你可能感兴趣的:(BUG,python,pytorch)