win10下最好用的pytorch安装教程

废话不多说我们直接进入正题。

1、首先确认你有无anaconda，有则跳过这一步，没有看我这篇博客安装anaconda，链接：https://blog.csdn.net/JohnLeeK/article/details/99582355，然后换源这样可以加快包的下载速度，参考我的这篇博客，链接：https://blog.csdn.net/JohnLeeK/article/details/102686424

2、点击这个链接进入pytorch官网，https://pytorch.org/get-started/locally/，如图我们按照需求来选择安装。

我安装的是1.2.0的版本，点击图中红色方框链接，然后我们找对应版本，这里安装可选pip的方式还有conda的方式.。如果安装1.5的版本直接复制图中的命令即可（安装1.2.0版本的往后看，跳过1.5版本）：

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

这里我选择用conda的方式安装（图二红色方框命令），因为pip的方式有点慢（选择图三红色方框命令即可），复制粘贴2中第一幅图红色方框内容，要注意的是我选的的cuda10.0的版本，这里需要注意一个事，打开你命令行窗口（win+r），执行这个命令：

nvidia-smi

结果如下

可以看到我我的驱动版本是417.88，只要你的驱动版本大于410，就安装cuda10.0的版本或者更版本都没问题

3、我们先在conda环境中创建一个pytorch的虚拟环境（命令行模式，cmd，快捷键win+r）：

conda create -n pytorch python=3.7

然后激活

activate pytorch

安装pytorch

然后我们执行这个命令（图二红色方框复制过来的命令）：

conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch

4、耐心等待一会儿，完成安装后我们检查pytorch是否安装成功

我们在conda的pytorch虚拟环境中输入：

python

然后输入如下代码

import torch

print(torch.cuda.is_available())

print(torch.__version__)

结果如下，出现这样的结果就代表的pytorch环境安装好啦！

可以看到我们安装的是GPU版本，然后版本为1.2.0。