win10+pytorch1.4+cuda10.1安装：从显卡驱动开始

先说一下，当前现状

• win10 64位，GPU NVIDIA GeForce 940M（已装驱动，但是该驱动最高只支持cuda10.0，需要卸载重装）
• 当前已有深度学习环境：anaconda3+pytorch1.0.1+cudatoolkit10.0+cuda10.0+cudnn10.0

因为我需要将pytorch模型部署到android上去（只有pytorch1.3以上版本支持pytorchMobile部署，所以不得不升级），当前最新版本为pytorch1.4+cudatoolkit10.0(pytorch1.4不支持cudatoolkit10.0，所以升级cudatoolkit势在必行）(另外提一下支持cudatoolkit10.0的最高pytorch版本为pytorch1.2)，博文大概内容如下：

• 检查显卡支持的cuda，安装最新驱动
• 安装pytorch1.4+cudatoolkit10.1
• 安装cuda10.1+cudnn10.1

相关概念：

• anaconda是一个管理python包、IDE和虚拟环境的一个IDE，而且包含了很多科学计算的包，很方便
• pytorch如果使用GPU，cudatoolkit是必须的
• 显卡驱动是GPU必备的，它会告知你所能支持的最高版本cuda
• NVIDIA 的CUDA(就是cuda)是GPU的并行计算框架，但光有框架还不行，还必须有针对深度学习的库，cudnn就是这个库
• CUDNN是NVIDIA打造的针对深度神经网络的加速库，是一个用于深层神经网络的GPU加速库。
• 所以如果你想搭建深度学习环境，必须先保证显卡驱动支持，安装CUDA、CUDNN，以及pytorch相关的环境，而且网上说要求cudatoolkit,cuda,cudnn的版本要保持一致

安装pytorch1.4+cudatoolkit10.1

建议使用anaconda，anaconda可以很好的管理python包、环境等，这里不再介绍安装过程，自己可以找一下，网上很多（安装完成之后，记得换源），这里有一些anaconda的常用命令：

conda list //显示已安装的python包，如果是通过pip安装的包，不会显示
conda search xxx //搜索python包
conda install xxx=1.2 //安装指定版本python包
conda install D:xxx //安装本地python包（绝对路径）
conda create -n pytorch1.4 python=3.7 //创建虚拟环境pytorch1.4，python版本为3.7
conda activate pytorch1.4 //启用虚拟环境
conda deactivate //退出当前虚拟环境
conda remove -n pytorch1.4 --all //删除虚拟环境

进入pytorch官网：https://pytorch.org

输入安装命令进行安装pytorch：conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
注：如果下载失败，会有错误提示：xxx包下载错误，复制其后的下载链接，用浏览器下载，然后进行本地安装（最好能科学上网）
安装完成之后进行验证：

import torch   # 没报错说明pytorch安装成功
print(torch.version.cuda) # 查看cuda版本
print(torch.cuda.is_available()) # 查看显卡是否可用，True表示可用

注：如果你还未安装cuda和cudnn，此时还不能支持pytorch进行深度学习

安装驱动

一般win10会自带NVIDIA驱动，所以你要先检查当前驱动所能支持的最高cuda版本，检查方式：在桌面，右键->选择“NVIDIA 控制版面”->左下角“系统信息”(这是更新驱动之后的截图)

下载驱动：https://www.geforce.cn/drivers

注：Notebooks才是是笔记本的驱动，之后下载最新的文件（文件是exe的，下载完成直接双击安装，如果你你已经有驱动了，它会帮你卸载旧版本安装新版本的驱动）。重启电脑之后，再检查一下驱动所能支持的最高cuda版本

驱动更新完成之后，让我惊呆了！！！：

我什么都没干，只是下载了python1.4和更新了驱动，就同时支持了？？？
要知道此时：

• 虚拟环境pytorch：python3.6+cudatoolkit10.0
• 虚拟环境pytorch1.4：python3.7+cudatoolkit10.1
• 显卡驱动最高支持cuda10.2
• 已安装了cuda10.0+cudnn10.0

我不知道为什么cuda10.0+cudnn10.0可以支持python3.7+cudatoolkit10.1，不是说cudatoolkit，cuda，cudnn版本必须保持一致吗？？，但这毕竟是好事…（但我已经下载了cuda10.1+cudnn10.1…）

下载cuda10.1+cudnn10.1

下载cuda10.1，官网：https://developer.nvidia.com/cuda-zone

注：等待完成，双击默认安装就可以了
下载cudnn10.1，官网：https://developer.nvidia.com/rdp/cudnn-download（要先注册账号，才能下载）

注：cudnn是一个压缩包
临门一脚，配置环境变量：（就这4个，配置到系统环境变量path里）

最后一步：将cudnn解压，复制cuda目录下所有文件，粘贴到CUDA的安装目录C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1文件夹下，至此就完成了显卡驱动、cuda、cudnn以及pytorch1.4的安装了。
我暂时没有安装cuda10.1，我想试试cuda10.0是不是真的支持pytorch1.4+cudatoolkit10.1，若有坑，会及时更新。