windows安装pytorch完全指南（以机械革命s1为例MX150显卡）

机械革命s1安装cuda

走了很多弯路啊啊啊

第一步，明确自己的显卡是什么，查看自己的显卡的驱动版本是什么

https://en.wikipedia.org/wiki/CUDA#GPUs_supported

以mx150显卡为例，通过表格找到算力为6.1，架构为pascal核心

然后找到满足这个算力和架构的cuda版本

我这里选择的cuda9.0版本，使用其他满足要求的版本应该也是可以的。

第二步，然后安装对应的驱动，我这里安装的驱动是cuda9.0

CUDA Toolkit 9.0 Download - Archived | NVIDIA Developer

安装完毕后使用命令nvcc –version显示正确信息则安装驱动成功

第三步，安装完驱动之后就是要找到对应合适版本的pytorch和对应版本的torchvision了，如果没安装正确的版本会使得torch无法使用GPU进行加速。

Previous PyTorch Versions | PyTorch

在该网站上找到对应的安装命令，可以看到对应的pytorch版本和torchvision版本。

注意：这里我们不必安装cudatoolkit因为已经安装完驱动了。

执行命令后网速太差了，经常下载断断续续。

第四步，网络加速，使用清华源，但是下载下来的文件不是whl格式，无法正确识别安装。于是考虑离线pip安装。

https://download.pytorch.org/whl/torch_stable.html

在这个网站上找到对应的pytorch和torchvision安装包，下载，使用pip离线进行安装，成功。

第一次写文章。。。求赞