No module named 'torchvision.models.feature_extraction’解決辦法

解決辦法一：

首先有這樣幾種可能，是因為所用的torch和torch vision版本不兼容，或者版本過低，可以將原來的torch和torchvision進行卸載然后重新安裝。
注意：一定要版本兼容的才可以

卸載指令：

1、使用conda卸載pytorchconda uninstall pytorchconda uninstall libtorchconda uninstall torchvision2、使用pip卸載pytorchpip uninstall torchpip uninstall libtorchpip 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是我的虛擬環境。
完成上述操作后，該問題即可解決。

總結

以上是生活随笔為你收集整理的No module named ‘torchvision.models.feature_extraction‘的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。