win10系統下的pytorch安裝及環境配置
阿新 • • 發佈:2019-02-18
對於想用pytorch的同學,官網只提供了兩種平臺下的安裝方式,linux和mac os,並沒有windows。
然而對於懶得重灌系統,又想用cuda加速的同學,就真的沒有其他辦法了嗎。不,其實windows下也是可以安裝ptorch的。
整個安裝分為四步走,分別是Anaconda3安裝、cuda toolkit安裝、pytorch安裝、torchvision安裝。
1、Anaconda3安裝
pytorch官網都推薦conda作為包管理工具,在win平臺下我們同樣要基於conda進行安裝。
安裝過程中可以選擇將Anaconda加入環境變數。加入環境變數的話,可以在cmd或者powershell下直接使用anaconda已經安裝好的包如ipython、spyder等。沒加環境變數,可以使用Anaconda prompt作為代替(實測執行速度沒差)。如果開始沒加,後來又想加的,自行搜尋win10環境變數新增,將一下幾行加入PATH(路徑就是你的安裝目錄)。建議加一下,省事!
路徑\Anaconda3
路徑\Anaconda3\Scripts
路徑\Anaconda3\Library\bin
2、cuda tookit安裝
下exe版本的,防止網速過慢。
下完之後一路裝就是了。
3、pytorch安裝
有了cuda和Anaconda,下一步就是裝pytorch了。進入cmd(沒加環境變數的同學進入Anaconda Prompt),根據註釋標註的版本輸入以下指令中的某一條。
# for CPU only packages conda install -c peterjc123 pytorch-cpu # for Windows 10 and Windows Server 2016, CUDA 8 conda install -c peterjc123 pytorch # for Windows 10 and Windows Server 2016, CUDA 9 conda install -c peterjc123 pytorch cuda90 # for Windows 7/8/8.1 and Windows Server 2008/2012, CUDA 8 conda install -c peterjc123 pytorch_legacy
我用的第三條,即win10和cuda90,木有問題。
4、torchvision安裝
cmd或者Anaconda Prompt進入當前目錄輸入以下指令:
pip install torchvision-0.1.8-py2.py3-none-any.whl
然後就大功告成啦!
怎麼看有沒有安裝成功呢?進入ipython輸入如下語句: