Tensorflow--WIN10系统1070安装GPU版本的Tensorflow1.8.0

这几天深度学习的过程中发现了一个问题，现在深度学习的过程中用到的tensorflow是CPU版本，而在训练的过程当中所用到的时间感觉特别长，因此想换成gpu版本的tensorflow，来缩短深度学习中模型训练的时间。

由于之前学习的过程当中所用到的tensorflow是1.0版本，所有写的代码都是基于1.0的版本，因此今天尝试安装的tensorflow是1.0版本的，基于python3.6

国际惯例，根据这篇博客，开始了安装的过程：

https://zhuanlan.zhihu.com/p/37086409

第一步根据步骤，点开我的电脑中的设备管理器，显示适配器中显示显卡为1070

接着上网百度搜索1070 SPECIFICATION

 

 出来的第一个网页点击进去之后，发现对应的cuda 后面写上对应的数量，因此来说应该是支持的

 接着确定tensorflow型号

打开cmd控制台

 输入(没有安装过tensorflow的这一步可以省略)

python

import tensorflow as tf

tf.__version__

结果显示我之前安装的cpu版本的tensorflow是1.2.0的，因此安装的gpu版本应该也要1.0的，不要到2.0，这里先记住，等下再进行搜索安装

接着点开刚刚英伟达官网的页面下载cuda

点击图中所示support按钮

 在对应的右边的区域有个技术支持，点击他之后 在对应的上边有对应的developers，选择cuda，然后选择下载就好了，在这里我下载对应的9.0版本--链接

这里附上详细安装步骤的网址：https://blog.csdn.net/a583621671/article/details/87272803

依次安装

 win10-1-2-3-4

安装时间挺长的，耐心等待之后

安装完成后，在控制台输入nvcc -V

如果出现以下的画面说明安装成功。

下载cudnn7.4.1：cudnn下载地址.

下载需要注册，注册完即可开始下载，之后解压文件，分别放置到

把其中的bin、include、lib中的文件分别复制到刚才cuda的安装路径：.NVIDIA GPU Computing ToolkitCUDAv9.0下的bin,include,lib中。

然后在Path添加以下的系统变量：(若不存在就添加)

C:Program FilesNVIDIA GPU Computing ToolkitCUDAv9.0in

C:Program FilesNVIDIA GPU Computing ToolkitCUDAv9.0include
C:Program FilesNVIDIA GPU Computing ToolkitCUDAv9.0libx64
C:Program FilesNVIDIA GPU Computing ToolkitCUDAv9.0extrasCUPTIlibx64
C:Program FilesNVIDIA GPU Computing ToolkitCUDAv9.0lib

接着将原本cpu版本的tensorflow覆盖为gpu版本

pip install --ignore-installed --upgrade tensorflow-gpu

如若想0安装特定版本的话则执行以下的语句：

pip install --ignore-installed --upgrade tensorflow_gpu==1.8.0

耐心等待其安装完成

测试能否用gpu运行卷积神经网络

 

运行速度极快