首先,安装python
PyTorch on Windows only supports Python 3.x; Python 2.x is not supported.
翻译:PyTorch在Windows上只支持Python 3.x;不支持Python 2.x。
可以通过下面几种方式安装:(一般直接到官网下载就行)
通过Package Manager下载Pytorch
而软件包管理器有两个可以选择:
- Anaconda
没有CUDA-capable系统或不需要CUDA:点击win图标-》找到Anaconda3-》打开Anaconda Prompt输入下面的命令
conda install pytorch torchvision cpuonly -c pytorch
有CUDA-capable系统或需要CUDA:键入
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
或者:
conda install pytorch torchvision cudatoolkit=9.2 -c pytorch -c defaults -c numba/label/dev
- pip(因为python里面自带了pip服务,所以你不需要再额外安装这个)
没有CUDA-capable系统或不需要CUDA:输入下面的命令
pip install torch==1.4.0+cpu torchvision==0.5.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
有CUDA-capable系统或需要CUDA:键入
pip install torch==1.4.0+cpu torchvision==0.5.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
或者:
pip install torch==1.4.0+cu92 torchvision==0.5.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html
验证是否安装完成
使用一段pytorch进行验证
- 打开cmd命令行
- 依次键入下面命令
python
from __future__ import print_function
import torch
x = torch.rand(5, 3)
print(x)
若打印出 类似以下数组,则为成功
tensor([[0.3380, 0.3845, 0.3217],
[0.8337, 0.9050, 0.2650],
[0.2979, 0.7141, 0.9069],
[0.1449, 0.1132, 0.1375],
[0.4675, 0.3947, 0.1426]])
可能出现的问题
- 使用Anaconda的conda命令进行安装无法成功下载所有文件
提示一下错误:
CondaHTTPError: HTTP 000 CONNECTION FAILED for url <https://conda.anaconda.org/pytorch/win-64/pytorch-1.4.0-py3.7_cpu_0.tar.bz2>
Elapsed: -
An HTTP error occurred when trying to retrieve this URL.
HTTP errors are often intermittent, and a simple retry will get you on your way.
建议采用pip命令进行安装:即上文的
pip install torch==1.4.0+cpu torchvision==0.5.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
#可能原因是:pip连接的下载服务器和使用conda命令的不同,而conda连接到的无法下载导致。
#安装软件尽量到官网下载并按其步骤执行,避免走太多弯路。