安装|深度学习环境配置Win10+CUDA+cuDNN+Tensorflow2.0+PyTorch1.2+Python3.7.6
系统环境：Win10
Python版本：3.7.6
CUDA版本：10.0
cuDNN版本：7.6.5
Tensorflow-gpu版本：2.0.1
PyTorch版本：1.2.0
深度学习环境配置Win10+CUDA+cuDNN+Tensorflow2.0+PyTorch1.2+Python3.7.6
- 前言
- 版本对应
- 查驱动版本
- CUDA
- 一、安装CUDA 10.0
-
- CUDA卸载（本人因为兼容问题需要重装CUDA）
- 二、安装cuDNN 7.6.5
-
- 1.下载包（前提需要注册一个免费的NVIDIA）
- 2.解压缩，复制文件
- 三、Tensorflow2.0.1配置
- 四、PyTorch配置
-
- 下载安装包到本地
- 实验
- 遇到问题
- 总结
前言 大四了，快毕业，毕业设计有关深度学习，寒假想在家跑点深度学习模型，只能用自己GTX1060笔记本了。自己的笔记本是Win10系统，也不想换Ubuntu了，毕竟windows下各种办公学习软件，查论文用起来更方便。由此记录一下win10配置深度学习环境的过程。（GPU环境搭建）
我的配置：
系统：win10 64位
python版本：3.7.6
tensorflow-gpu版本：2.0.1
CUDA版本：10.0
cuDNN版本：7.6.5。
注意事项
1、路径不要有中文。
2、Tensorflow版本、PyTorch版本、CUDA版本、cuDNN版本和python版本是一一对应的，你可能更新其中一个后就导致深度学习环境不能用了。
再次强调：Tensorflow版本、PyTorch版本、CUDA版本、cuDNN版本和python版本是一一对应的
版本对应 NVIDIA官网给出的官方信息
文章图片
??所有能安装配置的前提是电脑拥有NVIDIA显卡（N卡），需要根据显卡的驱动信息选择对应的版本！
查驱动版本 打开NVIDIA控制面板，查看面板中的系统信息，检查驱动版本。
文章图片
文章图片
也可以直接cmd命令行输入：
nvidia-smiCUDA CUDA11.2安装教程
一、安装CUDA 10.0 NVIDIA官网CUDA下载：10.0版本
【安装|深度学习环境配置Win10+CUDA+cuDNN+Tensorflow2.0+PyTorch1.2+Python3.7.6】??等待下载完成，NVIDIA因为是外网所以下载速度…真的是太慢了，但是你如果…“科学上网一下”会好很多…
文章图片
下载完成后是一个exe文件
文章图片
我安装的是默认路径
文章图片
点击 同意并继续(A)
文章图片
选择自定义安装，点击 下一步（有些东西我感觉不会用到，所以就自定义安装了）
文章图片
下一步选择想要安装的组件，因为我没有用过Visual Studio，所以去掉visual studio integration这一组件，Nsight我也不需要，这里解释一下，Visual Studio，Nsight都是CUDA C的IDE，如果不用的话可以都不用安装，我用的是Pycharm和Dev-C++，所以我就没装。
文章图片
这里直接默认，点击 下一步(N)即可
请注意：这里显示的是CUDA真正的安装位置
文章图片
等待安装完成
文章图片
安装完成，点击 关闭
文章图片
打开cmd，输入
nvcc -V验证安装是否成功，显示CUDA版本信息，说明CUDA安装成功
文章图片
CUDA卸载（本人因为兼容问题需要重装CUDA） 注意要留下：NVIDIA的图形驱动程序、NVIDIA Physx系统软件、NVIDIA GeForce Experience
如图所示：
文章图片
依次卸载即可，点击 卸载
文章图片
文章图片
文章图片
文章图片
卸载之后留存的 NVIDIA 控件
文章图片
使用软件管家删除注册表
文章图片
二、安装cuDNN 7.6.5
??cuDNN是基于CUDA的深度学习GPU加速库。cuDNN是专门为深度学习算法服务的。同一个深度学习算法用cuDNN+CUDA实现相比直接CUDA实现效率更高，运行速度更快，资源消耗更少。1.下载包（前提需要注册一个免费的NVIDIA） 注册账号就不多说了，直接注册就行，也可以直接微信扫码注册！
NVIDIA官网下载cuDNN: https://developer.nvidia.com/rdp/cudnn-archive
选择合适的版本
文章图片
之后就会让你登录！注册一个账号，登陆即可
文章图片
看到最上方有一个蓝条，就可以下载了
文章图片
2.解压缩，复制文件 现在完成后是一个zip文件，需要解压缩
文章图片
解压缩后得到三个文件夹，如下所示：
文章图片
??将解压后得到的的 bin ，include 和 lib 文件夹分别复制到cuda安装路径下与cuda的bin ，include 和lib文件夹合并。(相应文件夹下的文件复制粘贴)
??测试安装是否成功，运行CUDA的官方示例，在C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\extras\demo_suite路径下打开cmd，运行
.\deviceQuery.exe显示如下结果说明安装成功:
文章图片
三、Tensorflow2.0.1配置
文章图片
打开命令行，输入：
python -m pip install --upgrade pip
pip install tensorflow-gpu==2.0.1
等待安装结束即可
文章图片
#测试代码
import tensorflow as tf
gpu_ok = tf.test.is_gpu_available()
print("tf version:", tf.__version__)
print("use GPU", gpu_ok)
输出结果为：
文章图片
四、PyTorch配置
文章图片
访问PyTorch官网：
PyTorch官网https://pytorch.org/
往下翻有最新的安装代码，将NOTE里的代码复制到命令行中即可！
注意：一定注意自己的CUDA是否与自己本机配置相同，不同的可以进入下面的Previous versions of PyTorch访问往期下载命令
文章图片
我的是CUDA10.0，因此打开命令行，输入：
#conda安装
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch
#pip安装
pip install torch==1.2.0 torchvision==0.4.0 -f https://download.pytorch.org/whl/torch_stable.html
文章图片
粘贴代码到命令行：输入 y开始安装
文章图片
等待下载完成：
文章图片
下载完成：
文章图片
下载安装包到本地 ??这里需要提醒一下，PyTorch的版本在国内下载是非常非常非常慢的，因此，建议大家在Package一项中选择“pip”安装，然后在“Run this Command”这一栏，直接复制代码中的两个网址，在新的浏览器页面中直接输入：
举个例子：
https://download.pytorch.org/whl/cu100/torch-1.1.0-cp36-cp36m-win_amd64.whl （CUDA 10.0）
https://download.pytorch.org/whl/cu90/torch-1.1.0-cp36-cp36m-win_amd64.whl （CUDA 9.0）
首先，cd 到下载文件的目录，然后运行以下命令：
#安装
pip3 install pkg_name.whl#注意：将pkg_name换成下载的whl文件名(可能文件类型也不一样)
Pytorch 包下载链接
实验
import torch#导入pytorch
a = torch.cuda.is_available()#判断torch是否与cuda兼容
print(a)
ngpu= 1
device = torch.device("cuda:0" if (torch.cuda.is_available() and ngpu > 0) else "cpu")#判断是否是CUDA在运行
print(device)
print(torch.cuda.get_device_name(0)) #输出显卡型号
print(torch.rand(3,3).cuda())#随机张量的创建(二维)
文章图片
文章图片
遇到问题 ??我第一次下载完成后运行测试代码引入pytorch时报错，后来重新配置一下环境，类似问题消失！可能是下载完的包只是下载还没有配置好！
??我重新运行了一次安装代码，但是没有
-c pytorch，一定要删掉后面的-c pytorch，因为它代表直接从官网去下载，不会与本地安装好的匹配，还会全部重新下载。conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0
??之后它就会自动匹配之前离线安装好的东西，然后匹配相关的库的等级、版本之类的。
至此，win10下深度学习环境配置成功。
总结 此上就是深度学习环境的安装！！希望能帮助到大家。
推荐阅读
- 程序人生|程序员面试被问，有没有别家的offer（这个问题怎么回答？）
- java|Mybatis源码简析——实用框架必看
- 数据库|学习 MyBatis 的一点小总结 —— 底层源码初步分析
- dockerfile中telnet-server安装
- ★MATLAB算法仿真经验|【眼底检测】视网膜动静脉血管检测和特征计算matlab仿真
- python|「软件测试4」一文详解四大典型的白盒测试方法
- 全链路监控pinpoint安装部署
- 深度学习|李宏毅深度学习笔记 ---- 简介
- 机器学习|李宏毅机器学习GAN的笔记