操做系統:win10html
已安裝程序:Python 3.6 + Anaconda 5.1.0 + CUDA 9python
pytorch官網:https://pytorch.org/git
一、進入官網,從Get Started進入pytorch安裝教程,選擇對應操做系統、python版本、安裝方式、CUDA版本:github
可選擇多種安裝方式,在Run this Command可看到對應命令:this
一、Conda(Anaconda)操作系統
Command:.net
conda install pytorch torchvision -c pytorch
二、pip命令行
Command:3d
pip3 install https://download.pytorch.org/whl/cu90/torch-1.0.0-cp36-cp36m-win_amd64.whl pip3 install torchvision
//pip3不能用換成pip
三、Sourcecode
根據官網提示安裝,網址: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里正常使用。