win10系统下的pytorch安装及环境配置

对于想用pytorch的同学，官网只提供了两种平台下的安装方式，linux和mac os，并没有windows。

然而对于懒得重装系统，又想用cuda加速的同学，就真的没有其他办法了吗。不，其实windows下也是可以安装ptorch的。

整个安装分为四步走，分别是Anaconda3安装、cuda toolkit安装、pytorch安装、torchvision安装。

1、Anaconda3安装

pytorch官网都推荐conda作为包管理工具，在win平台下我们同样要基于conda进行安装。

下载地址https://www.anaconda.com/download/

安装过程中可以选择将Anaconda加入环境变量。加入环境变量的话，可以在cmd或者powershell下直接使用anaconda已经安装好的包如ipython、spyder等。没加环境变量，可以使用Anaconda prompt作为代替（实测运行速度没差）。如果开始没加，后来又想加的，自行搜索win10环境变量添加，将一下几行加入PATH（路径就是你的安装目录）。建议加一下，省事！

路径\Anaconda3  
路径\Anaconda3\Scripts  
路径\Anaconda3\Library\bin

2、cuda tookit安装

下载地址https://developer.nvidia.com/cuda-downloads

下exe版本的，防止网速过慢。

下完之后一路装就是了。

3、pytorch安装

有了cuda和Anaconda，下一步就是装pytorch了。进入cmd（没加环境变量的同学进入Anaconda Prompt），根据注释标注的版本输入以下指令中的某一条。

# for CPU only packages
conda install -c peterjc123 pytorch-cpu

# for Windows 10 and Windows Server 2016, CUDA 8
conda install -c peterjc123 pytorch

# for Windows 10 and Windows Server 2016, CUDA 9
conda install -c peterjc123 pytorch cuda90

# for Windows 7/8/8.1 and Windows Server 2008/2012, CUDA 8
conda install -c peterjc123 pytorch_legacy

我用的第三条，即win10和cuda90，木有问题。

4、torchvision安装

建议本地包安装，地址：https://pypi.python.org/pypi/torchvision/0.1.8

cmd或者Anaconda Prompt进入当前目录输入以下指令：

pip install torchvision-0.1.8-py2.py3-none-any.whl

然后就大功告成啦！

怎么看有没有安装成功呢？进入ipython输入如下语句：