no module named ‘torch._custom_ops‘

作者：繁依Fanyi0 | 2024-04-11 15:30:11

踩

pytorch缺少'torch._custom_ops'

报错分析
发现问题

报错分析

首先从原来出问题的代码去找问题，一路追踪到torchvision的init文件中出问题的部分。
在这里插入图片描述

发现问题

追踪到init文件发现是没有definition的，这时候我想到了是不是由于版本更新问题，之前的torch版本可能没有这个包，查找torch._custom_ops的文档，但是好像找不到，只找到了这个包的文件
https://fossies.org/linux/pytorch/torch/_custom_ops.py
这里发现使用的pytorch是2.2.1版本的

因此使用pip install --upgrade pytoch
然后uninstall torchvision重新下载即可

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/繁依Fanyi0/article/detail/405807