window下安装pytorch
2020-09-23 本文已影响0人
yimengtianya1
1、下载pytorch安装包
需要下载torch、torchvision两个安装包。
1.1 打开pytorch官网https://pytorch.org/
1.2 进入下载包界面
选择pip安装模式,语言版本,由于我电脑没有Nvidia的显卡,所以CUDA选择的None。如果有显卡的,查一下显卡适配的CUDA的版本,对应下载即可。
最后会出现安装命令:
pip install torch==1.6.0+cpu torchvision==0.7.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
图片.png
这里有两种安装方法:
(1)将安装包下载好了,再在cmd命令下,采用pip命令进行安装。
pip install [安装包文件路径]
(2)直接采用官网推荐的run this command里的命令进行安装。
pip install torch==1.6.0+cpu torchvision==0.7.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
这里,我采用的是第一种安装方案。主要是担心,直接安装网络下载比较慢,容易中途断掉,前功尽弃。
(3) 将图中的网址打开,可以看到对应的安装包。这里,对应安装命令里对应的版本,进行下载。
图片.png
我下载的版本是:
torch-1.6.0+cpu-cp36-cp36m-win_amd64.whl
torchvision-0.7.0+cpu-cp36-cp36m-win_amd64.whl
2、安装
打开cmd窗口,直接执行:
pip install [安装包路径地址]
先安装torch-1.6.0,再安装torchvision-0.7.0即可。
3、torch的windows依赖
torch在windows运行,依赖VC_redist.x64.exe。下载链接地址如下:
https://download.visualstudio.microsoft.com/download/pr/fd5d2eea-32b8-4814-b55e-28c83dd72d9c/952A0C6CB4A3DD14C3666EF05BB1982C5FF7F87B7103C2BA896354F00651E358/VC_redist.x64.exe
下载安装后,重启电脑即可。
4、验证安装
在IDE中执行以下代码,正常执行,即表示安装成功。若报错,可能是第3步安装的windows依赖有问题,按提示安装即可。
from __future__ import print_function
import torch
x=torch.rand(3,3)
print(x)