Win10使用pip安装PyTorch的方法简述

准备

确认配置

首先，要确认自己电脑是64位的还是32位的：

• 快捷键法：win+pause
  
• 我的电脑（此电脑） —— 右键 —— 属性
  

其次，要确认自己电脑的GPU的CUDA版本：

没有独立显卡的请在后面选择cpu版下载。

1. 进入NVIDIA控制面板（以下方法仅供参考¹，用其他方法能进入N卡控制面板也是可以的）
   1）Win10搜索并进入控制面板；（快捷键win+S）
   
   2）在控制面板右上角就搜索NVIDIA就能找到N卡的控制面板。
   
2. 点击帮助(H)-系统信息(I)；
3. 点击组件选项卡就可以看到CUDA版本了。

最后，确认自己安装的python版本：

使用win+R调出运行，并在运行中输入cmd，回车，在弹出的命令提示符窗口中输入以下语句：

	python --version #注意是两个-

然后就会显示当前使用的python版本，还没有安装python的可以看这里。

下载.whl包

1. 首先进入PyTorch的官网：pytorch.org
   点击Get Started。
2. 选择Windows，复制红线上的网址；
   
3. 把复制好的地址粘贴到浏览器地址框，并回车访问，可以看到一整个网页的超链接；
   
4. 使用Ctrl+F或F3调出查找框，并按如下规则找到需要的torch和torchvision（可以按第2步里官网推荐的版本）；
   
   示例：_{（CUDA10.1，torch1.5.0，python3.7，Windows，64位）}
   5.下载找到的链接，这里建议使用IDMan下载器（请自行百度下载）
   *基于IDMan下载器的下载方法：
   1）右键链接地址，点击复制链接地址(E)；
   2）进入IDMan，点击新建任务图标，将刚刚复制的链接粘贴进地址框，点击确定即可。

安装.whl包

找到刚刚下好的两个.whl文件所在的路径，使用win+R打开运行，输入cmd然后回车，来到命令控制符界面。
优先使用python -m pip install --upgrade pip命令更新一下自己的pip，待更新完成后使用

pip install 下载路径\***torch-***-cp**-cp**-win_amd64.whl
pip install 下载路径\***torchvision-***-cp**-cp**-win_amd64.whl

来安装本地whl包。
示例：_{（CUDA10.1，torch1.5.0，torchvision0.6.0，python3.7，Windows，64位，下载到E盘下）}

pip install E:\torch-1.5.0+cu101-cp37-cp37m-win_amd64.whl
pip install E:\torchvision-0.6.0+cu101-cp37-cp37m-win_amd64.whl

之前直接使用pip再加上清华源下载，发现torch只有低版本的，可能是镜像没同步的原因吧。心想不管它直接pip install torch的时候，又出现了如下错误：
ModuleNotFoundError: No module named ‘tools.nnwrap’
最后上S.O.搜了一下²才有了上面这个方案。

验证安装

使用win+R打开运行，输入cmd然后回车，来到命令控制符界面。
输入pip list³找找有没有如下两个项目和对应的版本号，有即为安装成功。

参考资料

---

1. Windows系统查看CUDA版本号——潇洒坤 ↩︎

2. ModuleNotFoundError: No module named ‘tools.nnwrap’ (windows)——CDJB, Harish Vutukuri ↩︎

3. Pip怎么查看已安装的包？教你查看pip已装模块——BinnLZeng ↩︎