站点图标 AI技术聚合

ModuleNotFoundError: No module named ‘torch.ao‘报错解决方案

提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档

文章目录

前言

在Python的GUI设计过程中,遇到了一个奇怪的问题“ModuleNotFoundError: No module named ‘torch.ao’”,一开始笔者以为这是由于缺少某个库所导致的原因,后来发现其实还是一个版本引发的问题。

一、解决方法

在Python的GUI设计过程中,又是需要使用torch和torchvision两个库。通过pip分别下载的torch和torchvision版本可能都是最新的版本。

但是,使用torch和torchvison的时候,其版本有着对应关系,如果版本不对,就可能会引发以上报错。所以解决的方法就是安装可以配套使用的torch和torchvison版本。

二、版本对应关系

在这里,笔者给出了torch和torchvison的对应版本关系:
torch和torchvison版本对应关系

总结

安装正确版本的torch和torchviosn即可解决问题。又是一个伤脑筋的版本问题。希望对大家有帮助,祝你们幸福!

文章出处登录后可见!

已经登录?立即刷新
退出移动版