1. 程式人生 > >windows下安裝torchvision報錯的解決方案

windows下安裝torchvision報錯的解決方案

環境 低版本 圖片 whl import cond dll 分享圖片 錯誤信息

今晚再次多次嘗試在win10下安裝pytorch以及torchvision,終於把上次的坑給填了。來來回回裝卸各種包有幾十遍了,稍微記錄一下,給後面遇到同樣問題的人一點提醒。
首先,介紹下安裝流程,比較簡單,在Anaconda Prompt下運行如下官方指定命令。(這裏用pip裝,pytorch也可以用conda裝,沒什麽區別,但torchvision只能用pip裝,原因下面有說)

pip3 install http://download.pytorch.org/whl/cpu/torch-0.4.1-cp36-cp36m-win_amd64.whl 
pip3 install torchvision

上次的坑主要是:安裝都成功了,並且import torch也沒問題,但是在import torchvision的時候,出現了如下所示的錯誤信息:DLL load failed:找不到指定模塊。
技術分享圖片


首先,我們得知道torchvision在conda裏是裝不了的,因為根本沒有提供給win的資源,因此我們只能使用pip安裝。
技術分享圖片
安裝完後,我們發現torchvision其實是有環境依賴的,其中有一條就是pillow的版本,而此次出錯的問題主要就是這個PIL的問題。
技術分享圖片
我在網頁上找資料的時候,沒有找到直接的解決辦法,但是有個博主在使用cv這個包的時候遇到了和我差不多的問題,最後他是通過把Pillow包的版本降到4.0.0後得以解決,因此我試著模仿,發現安裝torchvision的時候自動給我更新到5.2.0,最後我試著把版本降回到4.1.1,也就是能夠成功安裝的最低版本,竟然得以解決。

pip uninstall Pillow
pip install Pillow==4.1.1

因此最後得出的結論是:最新的Pillow包缺少了支持torchvision的東西,大家在遇到相同問題的時候,不妨把Pillow的版本降一降,希望能有所幫助。>.>

windows下安裝torchvision報錯的解決方案