1. 程式人生 > >windows平臺Pytorch安裝與使用

windows平臺Pytorch安裝與使用

Pytorch自0.4.0版本提供了windows系統平臺的支援。

自官網介紹的命令進行安裝,速度緩慢,可按照如下兩種方式進行安裝(已在之前安裝tensorflow時配置好了cuda和cudnn):

方法一:pip安裝

Python3.6+pip安裝cpu版本

pip install http://download.pytorch.org/whl/cpu/torch-0.4.0-cp36-cp36m-win_amd64.whl 

Python3.5+pip安裝cpu版本

pip install http://download.pytorch.org/whl/cpu/torch-0.4.0-cp35-cp35m-win_amd64.whl

Python3.6+pip安裝gpu版本
目前gpu版本支援cuda8.0,cuda9.0cuda9.1,請選擇對應的版本下載安裝,不要同時執行下面三個命令!

pip install http://download.pytorch.org/whl/cu80/torch-0.4.0-cp36-cp36m-win_amd64.whl  
pip install http://download.pytorch.org/whl/cu90/torch-0.4.0-cp36-cp36m-win_amd64.whl 
pip install http://download.pytorch.org/whl/cu91/torch-0.4.0-cp36-cp36m-win_amd64.whl 

Python3.5+pip安裝gpu版本
目前gpu版本支援cuda8.0,cuda9.0cuda9.1,請選擇對應的版本下載安裝,不要同時執行下面三個命令!

pip install http://download.pytorch.org/whl/cu80/torch-0.4.0-cp35-cp35m-win_amd64.whl
pip install http://download.pytorch.org/whl/cu90/torch-0.4.0-cp35-cp35m-win_amd64.whl
pip install http://download.pytorch.org/whl/cu91/torch-0.4.0-cp35-cp35m-win_amd64.whl

方法二:Conda安裝

如果你是Anaconda|Python使用者,就不需要區分Python3.5Python3.6,執行命令:

conda install pytorch -c pytorch 

就可以完成安裝。不過這個預設安裝的是cuda8.0gpu版本,如果你需要安裝cuda9.0cuda9.1.0gpu版本,請執行:

conda install pytorch cuda90 -c pytorch 

或者

conda install pytorch cuda91 -c pytorch 

來進行安裝。

測試安裝是否成功

import torch
print(torch.__version__)

如果輸出0.4.0,那麼恭喜Windows下的PyTorch0.4.0安裝成功!

最後需要安裝 torchvision

pip install torchvision