admin管理员组文章数量:1650829
No module named 'torchvision.models.feature_extraction’解决办法
解决办法一:
首先有这样几种可能,是因为所用的torch和torch vision版本不兼容,或者版本过低,可以将原来的torch和torchvision进行卸载然后重新安装。
注意:一定要版本兼容的才可以
卸载指令:
1、使用conda卸载pytorch
conda uninstall pytorch
conda uninstall libtorch
conda uninstall torchvision
2、使用pip卸载pytorch
pip uninstall torch
pip uninstall libtorch
pip uninstall torchvision
安装指令
安装pytorch前必须保证安装了cuda和cudnn
可以通过nvcc -V 查看自己的conda版本去pytorch进行安装
nvcc -V后的界面
然后去PyTorch官网找到适应的版本
https://pytorch/get-started/previous-versions/
如果还是没有解决,那就说明这个版本的torchvision包里面缺少了这一项,
看下面的解决办法二:
解决办法二:
可以在官网找到相应的代码,进行手动安装。
相应的代码链接:
https://pytorch/vision/main/_modules/torchvision/models/feature_extraction.html
然后按照代码中已有的模式进行学习装配
在models文件夹下面我们可以看到他封装好的库,我们也进行模仿学习把缺少的补回来。
按照模板,我们创建一个feature_extraction.py文件,将上面官网中的代码复制进.py文件,然后找到库所在的目录,是在你当前的虚拟环境下,我的如下:
D:\Anaconda3\envs\pytorch1.8\Lib\site-packages\torchvision\models
pytorch1.8是我的虚拟环境。
完成上述操作后,该问题即可解决。
本文标签: namedModuleTorchVisionfeatureextractionModels
版权声明:本文标题:No module named ‘torchvision.models.feature_extraction‘ 内容由热心网友自发贡献,该文观点仅代表作者本人, 转载请联系作者并注明出处:https://m.elefans.com/dongtai/1729531690a1204888.html, 本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容,一经查实,本站将立刻删除。
发表评论