Windows下安装PyTorch

PyTorch进行神经网络的学习十分有用, 但是,其在*的安装包下载十分缓慢。这里介绍一下我在Windows10中安装PyTorch的过程与建议。

1、版本选择,进入PyTorch的官网:https://pytorch.org/get-started/locally/

Windows下安装PyTorch

这里需要提醒一下,PyTorch的版本在国内下载是非常非常非常慢的,本人刚开始也是尝试了无数次的官网直接下载,最终都因为下载速度无比的慢而失败。以上图为例,如果小伙伴们想尝试直接官网下载,以管理员运行cmd, 输入如下命令:

pip install torch==1.5.0+cpu torchvision==0.6.0+cpu -f https://download.pytorch.org/whl/torch_stable.html

2、下载安装包到本地

建议大家在Package一项中选择“pip”安装,然后在“Run this Command”这一栏,直接复制代码中的两个网址,

以我上面的版本选择为例,即在新的浏览器页面中直接输入:

https://download.pytorch.org/whl/torch_stable.html

选择cpu/torch==1.5.0  cpu/torchvision==0.6.0的版本下载到本地,浏览器下载速度较慢,迅雷会稍微快点。

3、安装(这里我直接用本地下载的源失败了,然后加上清华源成功了)

以管理员身份运行cmd
输入如下代码:(红色部分是你下载的安装包的存放路径,蓝色部分是你下载的安装包名称)

pip install C:\Users\XXX\Desktop\torch-1.5.0+cpu-cp38-cp38-win_amd64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple/

再输入代码(红色部分是你下载的安装包的存放路径,蓝色部分是你下载的安装包名称):

pip install C:\Users\XXX\Desktop\torchvision-0.6.0+cpu-cp38-cp38-win_amd64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple/

4、卸载Pytorch
以管理员身份运行cmd
输入如下代码:

pip uninstall torch

再输入如下代码:

pip uninstall torchvision

5、测试包安装成功
以管理员身份运行cmd
输入:python
输入如下代码:

from __future__ import print_function
import torch
x=torch.rand(5,3)
print(x)
Windows下安装PyTorch