No module named ‘torchvision.models.feature_extraction‘

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
1
2
3
4
5
6
7
8
9
10

安装指令

安装pytorch前必须保证安装了cuda和cudnn
可以通过nvcc -V 查看自己的conda版本去pytorch进行安装

nvcc -V后的界面

然后去PyTorch官网找到适应的版本
https://pytorch.org/get-started/previous-versions/

如果还是没有解决，那就说明这个版本的torchvision包里面缺少了这一项，
看下面的解决办法二：

解决办法二：

可以在官网找到相应的代码，进行手动安装。

相应的代码链接：
https://pytorch.org/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是我的虚拟环境。
完成上述操作后，该问题即可解决。