Windows下安装PyTorch
PyTorch进行神经网络的学习十分有用, 但是,其在*的安装包下载十分缓慢。这里介绍一下我在Windows10中安装PyTorch的过程与建议。
1、版本选择,进入PyTorch的官网:https://pytorch.org/get-started/locally/
这里需要提醒一下,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)