ModuleNotFoundError: No module named ‘torchvision.models.feature_extraction‘ 解决

一看就是版本太老的问题,但是难就难在把这个升级了,其他的包怎么兼容呢?特别是和pytorch有关的就相当麻烦,因为我用的是gpy版本的pytorch,还需要考虑显卡的版本,所以就更可怕了

升级到新版

(这里就不写了,我现在还不会)

在缺失的地方手动把源代码贴进去

错误是找不到feature_extraction.py这个文件

找到该文件根目录地址

使用语句:

import module
print(module.__file__)

/home/algroup/anaconda3/envs/env1/lib/python3.7/site-packages/torchvision/models/__init__.py

把源代码贴进去

在路径/home/algroup/anaconda3/envs/env1/lib/python3.7/site-packages/torchvision/models下创建文件feature_extraction.py,把从pytorch官网找到的源代码贴进去;
完事!

参考

查看anaconda安装库/包的位置(路径)
No module named ‘torchvision.models.feature_extraction‘

你可能感兴趣的:(Debug,pytorch)