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

安装指令

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

nvcc -V后的界面
No module named ‘torchvision.models.feature_extraction‘_第1张图片
然后去PyTorch官网找到适应的版本
https://pytorch.org/get-started/previous-versions/
No module named ‘torchvision.models.feature_extraction‘_第2张图片
如果还是没有解决,那就说明这个版本的torchvision包里面缺少了这一项,
看下面的解决办法二:

解决办法二:

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

相应的代码链接:
https://pytorch.org/vision/main/_modules/torchvision/models/feature_extraction.html
No module named ‘torchvision.models.feature_extraction‘_第3张图片

然后按照代码中已有的模式进行学习装配

No module named ‘torchvision.models.feature_extraction‘_第4张图片

在models文件夹下面我们可以看到他封装好的库,我们也进行模仿学习把缺少的补回来。
No module named ‘torchvision.models.feature_extraction‘_第5张图片
按照模板,我们创建一个feature_extraction.py文件,将上面官网中的代码复制进.py文件,然后找到库所在的目录,是在你当前的虚拟环境下,我的如下:

D:\Anaconda3\envs\pytorch1.8\Lib\site-packages\torchvision\models

pytorch1.8是我的虚拟环境。
完成上述操作后,该问题即可解决。

你可能感兴趣的:(深度学习中的问题,pytorch,深度学习,缺少库)