admin管理员组

文章数量:1650774

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://www.elefans.com/dongtai/1729531690a1204888.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。