天天看點

win10系統下的pytorch安裝及環境配置

對于想用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版本的,防止網速過慢。

win10系統下的pytorch安裝及環境配置

下完之後一路裝就是了。

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

win10系統下的pytorch安裝及環境配置

cmd或者Anaconda Prompt進入目前目錄輸入以下指令:

pip install torchvision-0.1.8-py2.py3-none-any.whl
           

然後就大功告成啦!

怎麼看有沒有安裝成功呢?進入ipython輸入如下語句:

win10系統下的pytorch安裝及環境配置

繼續閱讀