Tensorflow教程02-Windows 10下安装tensorflow 1.5.0 GPU with Anaconda

禁止一切形式未经作者同意的转载。

  • 2018/2/21: 由于Anaconda 更新原命令activate tensorflow 改为 conda activate tensorflow
  • 2018/2/12: 感谢评论,已修改关于python 3.6的描述。经过测试windows with anaconda 可以使用python 3.6,需要注意,如使用3.6,则在创建conda环境时需使python=3.6,详情见原文。
  • 2018/2/11:关于CUDA和cuDNN版本,目前官网更新为CUDA9和cuDNN6,我之前实测CUDA9和cuDNN7完美运行,CUDA9和cuDNN6大家可以试一下。

系列教程将持续更新,最近可能慢一些,实验室Boss催实验催的紧,以后关于复杂网络构建可能会做视频,码字太累。。

TensorFlow 1.5.0 现已公开,如果您在Windows或Linux上使用GPU加速,TensorFlow 1.5现在已经支持CUDA 9和cuDNN 7。本文就来为大家详细地介绍一下如何在Windows下安装TensorFlow GPU版本。本教程基于Anaconda 而不是pip安装TensorFlow,主要原因由于TensorFlow安装在conda虚拟环境的不会对系统python产生影响,可安装多种不同python版本。

首先下载和安装Anaconda Python 3.6 version,关于获得Python 3.5或其他Python版本,之后会提到,并且TensorFlow Windows版目前官方支持Python 3.5和Python 3.6. 安装时无脑下一步即可,不必理会其他选项。

Downloadswww.anaconda.com《Tensorflow教程02-Windows 10下安装tensorflow 1.5.0 GPU with Anaconda》

然后根据自己GPU型号安装最新Nvidia GPU驱动:

NVIDIA 驱动程序下载www.nvidia.cn

目前官方文档显示为需CUDA 9.0 和 cuDNN 6。实际已经迁移到CUDA 9.0 和 cuDNN 7。安装CUDA 9.0:

CUDA Toolkit 9.0 Downloadsdeveloper.nvidia.com

  1. 选择自己的操作系统版本
  2. 选择Installer Type : exe(local)
  3. 选择Base Installer
  4. 下载后双击安装
  5. 安装选项选择自定义,只勾选CUDA如图1
  6. 之后一直点击下一步即可

《Tensorflow教程02-Windows 10下安装tensorflow 1.5.0 GPU with Anaconda》
《Tensorflow教程02-Windows 10下安装tensorflow 1.5.0 GPU with Anaconda》 图1

安装cuDNN 7(目前cuDNN 6 也可以,官网文档最近更新):

NVIDIA cuDNNdeveloper.nvidia.com《Tensorflow教程02-Windows 10下安装tensorflow 1.5.0 GPU with Anaconda》
《Tensorflow教程02-Windows 10下安装tensorflow 1.5.0 GPU with Anaconda》
《Tensorflow教程02-Windows 10下安装tensorflow 1.5.0 GPU with Anaconda》 图2
《Tensorflow教程02-Windows 10下安装tensorflow 1.5.0 GPU with Anaconda》
《Tensorflow教程02-Windows 10下安装tensorflow 1.5.0 GPU with Anaconda》 图3
《Tensorflow教程02-Windows 10下安装tensorflow 1.5.0 GPU with Anaconda》
《Tensorflow教程02-Windows 10下安装tensorflow 1.5.0 GPU with Anaconda》 图4
《Tensorflow教程02-Windows 10下安装tensorflow 1.5.0 GPU with Anaconda》
《Tensorflow教程02-Windows 10下安装tensorflow 1.5.0 GPU with Anaconda》 图5

  1. cuDNN7 是一个压缩包,下载后解压到某个地方,比如D盘,如图2所示。
  2. 搜索环境变量如图3,可以百度如何修改系统环境变量。之后点击环境变量(图4)。
  3. 之后选择Path,编辑,新建,将D:\cudnn-9.0-windows10-x64-v7\cuda\bin添加到环境变量,如果你没有解压到D盘请替换路径。至此cuDNN安装完成。

安装TensorFlow 1.5.0 GPU版本 :按以下步骤在Anaconda环境中安装TensorFlow:(输入命令后需回车运行)NVIDIA cuDNN安装TensorFlow 1.5.0 GPU版本 :按以下步骤在Anaconda环境中安装TensorFlow:(输入命令后需回车运行)

  • 打开Anaconda Prompt通过调用以下命令创建一个名为tensorflow_gpu的conda环境:
conda create -n tensorflow_gpu pip python=3.5
  • tensorflow_gpu 可以换成大家自己的名字,python=3.5代表这个虚拟环境的python版本,如果大家想使用python 3.6,就把python=3.5 替换为 python=3.6即可,输入后回车,一段时间后显示Proceed ([y]/n)?,输入y,回车,结果如图6。

《Tensorflow教程02-Windows 10下安装tensorflow 1.5.0 GPU with Anaconda》
《Tensorflow教程02-Windows 10下安装tensorflow 1.5.0 GPU with Anaconda》 图6

  • 通过发出以下命令来激活名为tensorflow_gpu的conda环境:(回车后会有类似图7的改变)
conda activate tensorflow_gpu

《Tensorflow教程02-Windows 10下安装tensorflow 1.5.0 GPU with Anaconda》
《Tensorflow教程02-Windows 10下安装tensorflow 1.5.0 GPU with Anaconda》 图7

  • 用以下命令,在名为tensorflow_gpu的conda环境中安装GPU版本TensorFlow。要安装TensorFlow的GPU版本,请输入以下命令回车运行后等待安装:
pip install --ignore-installed --upgrade tensorflow-gpu
  • 上个命令会安装最新版的TensorFlow,如果想安装指定版本如1.4.0版,请使用:
pip install tensorflow-gpu==1.4.0
  • 经过一段时间(1-5分钟)后会显示类似图8,表明TensorFlow GPU版已经安装成功:

《Tensorflow教程02-Windows 10下安装tensorflow 1.5.0 GPU with Anaconda》
《Tensorflow教程02-Windows 10下安装tensorflow 1.5.0 GPU with Anaconda》 图8

  • 我们可以通过一个简单的程序测试:在命令行中输入python,回车,进入到python(图9)。接着输入import tensorflow as tf ,若出现图10则说明安装成功。

《Tensorflow教程02-Windows 10下安装tensorflow 1.5.0 GPU with Anaconda》
《Tensorflow教程02-Windows 10下安装tensorflow 1.5.0 GPU with Anaconda》 图9
《Tensorflow教程02-Windows 10下安装tensorflow 1.5.0 GPU with Anaconda》
《Tensorflow教程02-Windows 10下安装tensorflow 1.5.0 GPU with Anaconda》 图10

  • 原生Python不利于代码编写和调试,所以我们需要一个IDE,打开Anaconda Navigator,如图11。在Environments下可以看到我们刚刚创建的tensorflow_gpu环境(图12)。

《Tensorflow教程02-Windows 10下安装tensorflow 1.5.0 GPU with Anaconda》
《Tensorflow教程02-Windows 10下安装tensorflow 1.5.0 GPU with Anaconda》 图11
《Tensorflow教程02-Windows 10下安装tensorflow 1.5.0 GPU with Anaconda》
《Tensorflow教程02-Windows 10下安装tensorflow 1.5.0 GPU with Anaconda》 图12

  • 切换到Home, 并将Application on改为tensorflow_gpu。在页面最后选择Spyder,点击install,如图13。安装完成后点击lanch,启动Spyder。

《Tensorflow教程02-Windows 10下安装tensorflow 1.5.0 GPU with Anaconda》
《Tensorflow教程02-Windows 10下安装tensorflow 1.5.0 GPU with Anaconda》 图13

  • 依次输入以下代码:如图14所示

《Tensorflow教程02-Windows 10下安装tensorflow 1.5.0 GPU with Anaconda》
《Tensorflow教程02-Windows 10下安装tensorflow 1.5.0 GPU with Anaconda》 图14

import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
  • 点击运行或按F5,之后点击run。会显示如图15的结果。

《Tensorflow教程02-Windows 10下安装tensorflow 1.5.0 GPU with Anaconda》
《Tensorflow教程02-Windows 10下安装tensorflow 1.5.0 GPU with Anaconda》 图15

至此,大家就可以使用Tensorflow GPU版啦。

码字辛苦,欢迎打赏。

    原文作者:赵超
    原文地址: https://zhuanlan.zhihu.com/p/33709404
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞