PyTorch在Windows10的安装步骤

首先,安装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

而软件包管理器有两个可以选择:

  1. 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
  1. 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进行验证

  1. 打开cmd命令行
  2. 依次键入下面命令
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]])

过程截图

可能出现的问题

  1. 使用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连接到的无法下载导致。
#安装软件尽量到官网下载并按其步骤执行,避免走太多弯路。

参考链接:

1.Pytorch官网
2.forums讨论区

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值