Windows系统下官方正式版PyTorch 0.4.0 安装测试

0.引言

喜大普奔，PyTorch于近日终于正式发布官方Windows版本啦！看到消息后马上进行了安装和测试，在这里记录一下。由于官方在windows系统上目前暂时只支持python3版本，所以今天将在Windows10上通过Anaconda3+python3.6进行安装和测试。

无论在Ubuntu还是在Windows中，我觉得Pytorch的安装一直是如此的简单明了，没有其他框架来的那么复杂，所以从安装上我就很喜欢Pytorch.

我这次安装PyTorch将按照如下顺序和版本进行安装：

• 1.安装Nvidia显卡驱动
• 2.安装CUDA9.1
• 3.安装cuDNN7.1
• 4.安装Anaconda3
• 5.安装PyTorch 0.4.0
• 6.多GPU实例测试

---

1.安装Nvidia 显卡驱动

在Windows上，基本都已安装有Nvidia的显卡驱动了，如果未安装或需要更新版本，可在Nvidia官网进行下载。下载完成后双击安装，路径默认即可。

---

2.安装CUDA9.1

2.1 下载

首先在英伟达官网下载CUDA9.1的安装包。

2.2 安装

下载本地版的exe可执行文件，完成后双击进行安装。安装将分别经历如下几个阶段：