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后的界面
在这里插入图片描述
然后去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是我的虚拟环境。
完成上述操作后,该问题即可解决。

  • 6
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值