windows上配置pytorch


操作系統:win10

已安裝程序:Python 3.6 + Anaconda 5.1.0 + CUDA 9

pytorch官網:https://pytorch.org/

1、進入官網,從Get Started進入pytorch安裝教程,選擇對應操作系統、python版本、安裝方式、CUDA版本:

可選擇多種安裝方式,在Run this Command可看到對應命令:

1、Conda(Anaconda)

Command:

conda install pytorch torchvision -c pytorch

2、pip

Command:

pip3 install https://download.pytorch.org/whl/cu90/torch-1.0.0-cp36-cp36m-win_amd64.whl
pip3 install torchvision
//pip3不能用換成pip

3、Source

根據官網提示安裝,網址:https://github.com/pytorch/pytorch#from-source,個人認為沒有前兩種方便,畢竟cmd下可直接安裝。

第一次我使用conda,在cmd命令行下安裝,輸入conda install pytorch torchvision -c pytorch,報錯:找不到對應的包,就轉為使用pip方法,后來查了一下,可以手動找包,參考Blog:https://blog.csdn.net/muyimo/article/details/84567630 和 https://blog.csdn.net/ksws0292756/article/details/79192268。(此種方法未經嘗試,僅供參考)

接下來說一說pip方法安裝:

進入anaconda prompt,運行command:pip install https://download.pytorch.org/whl/cu90/torch-1.0.0-cp36-cp36m-win_amd64.whl

下載后提示:exist pip 10, but pip 18.1 already exist,建議執行命令:python -m pip install --upgrade pip,應該是pip版本問題,所以我按建議更新了pip,此過程卸載了pip 10,安裝了pip 18.1。更新完pip后,重新執行命令,即提示安裝成功。再繼續運行第二條命令,成功。

 

檢查安裝是否成功:

cmd下進入python,依次輸入import torch和import torchvision,沒報錯即安裝成功

在這里可以看到import torch安裝成功,但torchvision安裝失敗。這里我再次pip install torchvision,出現以下提示:

 提示是已經成功安裝的。再次輸入import torchvision,還是提示相同的錯誤。查找之后發現,torchvision是有依賴環境的,就是上圖的這幾條:

而問題就出在第一條Pillow上:

在這里只要pillow版本>=4.1.1就可以,括號里標明了現在的版本是5.0.0,借鑒了有一個博主把pillow版本降到4.1.1的方法 http://www.mamicode.com/info-detail-2424578.html,執行pip install pillow==4.1.1:

把pillow版本降到4.1.1之后,真的成功了:

結論就是新版本的pillow缺少支持torchvision的東西,降版本可以解決這個問題。

 

 

其實可以直接進入cmd,使用pip方法安裝,但我在cmd命令行下執行命令時,出現報錯:'pip'不是內部或外部命令,也不是可運行的程序。我重新將pip所在的文件路徑加入系統變量PATH里(python路徑下的lib文件夾里:),還是找不到命令,於是轉為anaconda prompt里執行。在更新了pip之后,進入cmd,輸入pip -v檢查pip命令是否可用:

pip命令更新后就可在cmd里正常使用。

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM