對于想用pytorch的同學,官網隻提供了兩種平台下的安裝方式,linux和mac os,并沒有windows。
然而對于懶得重裝系統,又想用cuda加速的同學,就真的沒有其他辦法了嗎。不,其實windows下也是可以安裝ptorch的。
整個安裝分為四步走,分别是Anaconda3安裝、cuda toolkit安裝、pytorch安裝、torchvision安裝。
1、Anaconda3安裝
pytorch官網都推薦conda作為包管理工具,在win平台下我們同樣要基于conda進行安裝。
下載下傳位址https://www.anaconda.com/download/
安裝過程中可以選擇将Anaconda加入環境變量。加入環境變量的話,可以在cmd或者powershell下直接使用anaconda已經安裝好的包如ipython、spyder等。沒加環境變量,可以使用Anaconda prompt作為代替(實測運作速度沒差)。如果開始沒加,後來又想加的,自行搜尋win10環境變量添加,将一下幾行加入PATH(路徑就是你的安裝目錄)。建議加一下,省事!
路徑\Anaconda3
路徑\Anaconda3\Scripts
路徑\Anaconda3\Library\bin
2、cuda tookit安裝
下載下傳位址https://developer.nvidia.com/cuda-downloads
下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安裝
建議本地包安裝,位址:https://pypi.python.org/pypi/torchvision/0.1.8
cmd或者Anaconda Prompt進入目前目錄輸入以下指令:
pip install torchvision-0.1.8-py2.py3-none-any.whl
然後就大功告成啦!
怎麼看有沒有安裝成功呢?進入ipython輸入如下語句: