【保姆级教程】光速安装gpu版本pytorch

以上9点就是安装gpu版本pytorch全程，接下来一一详细解释

1.如何查看显卡版本？

点这个英伟达设置，顶部帮助→系统信息

找不到英伟达图标，也可以点右键我的电脑→管理→设备管理器→显示适配器，也能看到显卡型号

接下来去这个网站https://developer.nvidia.com/zh-cn/cuda-gpus#compute，点下面这个


如果可以找到自己的显卡型号，说明没问题，的确是可以安装CUDA的，接下来每一步都能走通

**然后更新或安装显卡驱动，**进这个网站www.nvidia.cn/drivers，

product type选GeForce，然后search，找到下面的选项

再点search，然后下载那个安装包，说明一下本文章的下载都不需要挂梯子！但是能不能成功开始下载，能不能快速下载完，看运气，我的时快时慢，而是下载失败，多试几次。
下载完后双击安装，全部默认选项，完成显卡驱动更新。注意安装包别删，一会儿还要用！

2.明确你的电脑能装什么版本CUDA

我的是GeForce系列的1050，在英伟达设置的系统信息里点组件

这个CUDA 11.2就是目前该电脑所能安装的最高版本CUDA。

3.明确目前Pytorch适配什么版本的CUDA

我们不一定要安装最高版本的CUDA，能用才是王道，pytorch目前最高好像只支持11.0版本的CUDA，所以先进这个网站
https://download.pytorch.org/whl/torch_stable.html

cu101/torch说明这是适配10.1版本的gpu版本pytorch，而且1.4.0的pytorch算比较新了，再加上我的python是3.6的，所以我们点中这个，并下载。推荐就下载10.1CUDA的，这是稳定版本

同理把torchvision也下载，torchvision是各种卷积神经网络模型.

这里需要注意的是torchvision和torch之间是有版本对应关系的,如下图:(源于https://github.com/pytorch/vision#installation)

4.下载安装CUDA

点这个网站https://developer.nvidia.com/cuda-downloads

查看以往版本CUDA

选择这个10.1版本

按照图中选项点好后下载，下载多久看运气，我前半小时都几十kb。后面突然十几MB/s，瞬间下载完成。然后安装，文件位置啥的全不要改。一切默认，进行安装，之后重启一下

5.重装显卡驱动

装完cuda可能会把显卡驱动删除，很迷惑，所以找到刚才那个显卡驱动更新包，双击重新安装

6.下载安装cuDNN

进入这个网站https://developer.nvidia.com/cudnn，点下载cuDNN，点join now注册一个账号，该填的就填，这里注册得填好几次信息，别乱点叉或不填啊，仔细看那些内容，最后会引导你去下载cuDNN，选适配10.1CUDA的

下载完解压，里面3个文件夹，1个文件。

然后打开文件夹位置C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1

这是刚才CUDA安装的位置，可以看到同样有这3个文件夹。
打开cuDNN里的bin，复制里面全部东西

再打开CUDA里的bin，粘贴

同样的把cudnn里的include和lib的文件也也全部复制到对应目录。
然后点CUDA里的extras，进入demo_suite文件夹
win+R键输入cmd，进入dos，然后输入cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\demo_suite，回车，再输入bandwidthTest

出现pass说明大功告成，你的电脑已经成功能用于gpu加速深度学习了，接下来只要安装对应gpu版本tensorflow或者pytorch即可。

8.新建环境，安装gpu版本pytorch

打开anaconda prompt，输入conda create with_gpu python=3.6创建名为with_gpu的新环境，关于环境的理解请参考https://blog.csdn.net/coolsmartboy/article/details/106436005。创建完成后输入指令activate with_gpu
然后打开之前pytorch和torchvision两个安装包的文件位置，复制文件位置C:\Users\74458\Desktop\FinalDesign，加上文件名torch-1.4.0-cp36-cp36m-win_amd64.whl，构成完整安装位置C:\Users\74458\Desktop\FinalDesign\torch-1.4.0-cp36-cp36m-win_amd64.whl，然后输入pip install C:\Users\74458\Desktop\FinalDesign\torch-1.4.0-cp36-cp36m-win_amd64.whl，同样的再把torchvision装上去

9.验证gpu版本pytorch是否安装成功

在anaconda prompt中activate with_gpu，再输入python，然后输入import torch，再输入print(torch.cuda.is_available())，输出true，安装成功。

觉得不错可以点个赞！还有安装问题的话，可以联系我付费远程解决。