windows平臺Pytorch安裝與使用
阿新 • • 發佈:2018-11-21
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.0
和cuda9.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.0
和cuda9.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.5
和Python3.6
,執行命令:
conda install pytorch -c pytorch
就可以完成安裝。不過這個預設安裝的是cuda8.0
的gpu
版本,如果你需要安裝cuda9.0
或cuda9.1.0
的gpu
版本,請執行:
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