window下安装pytorch

2020-09-23  本文已影响0人  yimengtianya1

1、下载pytorch安装包

需要下载torch、torchvision两个安装包。
1.1 打开pytorch官网https://pytorch.org/

图片.png
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)
上一篇下一篇

猜你喜欢

热点阅读