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运行卷积神经网络

 

运行速度极快

原文地址:https://www.cnblogs.com/halone/p/12582458.html