1. 程式人生 > >win10系統下的pytorch安裝及環境配置

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

對於想用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輸入如下語句: