Windows10下搭建TensorFlow环境

转载请注明源出处:http://www.cnblogs.com/lighten/p/6753695.html

这篇文章介绍了一下在Windows上安装TensorFlow的步骤,主要是翻译了一下官方的安装指导,官方指导网址:这里。(需要FQ访问)。

1 决定安装哪个版本的TensorFlow

你必须选择下列TensorFlow的其中一种类型进行安装(可以都装):

仅支持CPU版本的TensorFlow。如果你的系统中没有NVIDIA的GPU,你必须安装这个版本。注意这个版本通常是非常容易安装的,只需要5到10分钟,所以即使你有NVIDIA的GPU,我们也推荐你先装这个版本。

支持GPU版本的TensorFlow。通常TensorFlow的程序在GPU上运行比CPU快。因此,如果你的系统有满足以下所示的前提条件的NVIDIA的GPU,并且你所需要运行的应用很依赖性能,最终你应该安装这个版本。

2 运行GPU版本的TensorFlow所需条件

如果你使用本指南介绍的机制之一来安装具有GPU支持的TensorFlow,则系统上必须安装以下NVIDIA软件:

CUDA® Toolkit 8.0。有关详细信息,请参阅NVIDIA的文档确保将相关的Cuda路径名追加到%PATH%环境变量中,如NVIDIA文档所述。

与CUDA Toolkit 8.0相关的NVIDIA驱动程序。

cuDNN v5.1。有关详细信息,请参阅NVIDIA的文档。请注意,cuDNN通常安装在与其他CUDA DLL不同的位置。确保将你安装cuDNN DLL的目录添加到%PATH%环境变量中。

具有CUDA Compute Capability 3.0或者更高版本的GPU卡。有关支持的GPU卡的列表,请参阅NVIDIA文档

以上是官方原文,其实只需要下载两个软件,一个是CUDA8.0,另一个是cuDNN 5.1。第一个软件是一个exe文件,直接点开自动安装就可以了,环境变量会自动添加,在控制台输入:nvcc -V看到下图就是好了:

Windows10下搭建TensorFlow环境

第二个软件是一个压缩包,解压后放到任意一个文件夹,在系统的Path路径变量下添加路径就可以了。双击此电脑->系统属性->高级系统设置->环境变量->找到PATH变量编辑就可以了,具体如图:

Windows10下搭建TensorFlow环境

除了根目录,把bin的相关目录也要添加进去,这样这GPU所要的条件就完成了。此处参考文章。这篇文章也给了下载地址。

3 决定如何安装TensorFlow

你必须选择安装TensorFlow的机制。支持的选择如下:

"本地"pip

Anaconda

本地的pip在你的系统上直接安装TensorFlow,而不需要通过虚拟环境。由于本地pip安装不会在一个封闭的隔离容器中,所以pip安装可能影响你系统上其它基于Python的安装。然而,如果你了解pip和你的Python环境,本地pip安装通常只需要一个命令。并且,如果你使用本地pip安装,用户可以从系统上任何目录运行TensorFlow程序。

在Anaconda中,你可以使用conda来创建一个虚拟环境。但是,在Anaconda中,我们建议你使用pip install命令,而不是conda install命令。

注意:conda包是社区支持的,没有正式的支持。也就是说,TensorFlow团队既不测试也不维护这个conda包。使用这个包,你自己要承担风险。

4 使用本地的pip安装

如果你的计算机上未安装以下版本的Python,请立刻安装:

Python 3.5.x from python.org

TensorFlow在Windows上只支持3.5.x版本的Python。注意,Python 3.5.x自带了pip3软件包管理器,这是你将用于安装TensorFlow的程序。

要安装TensorFlow,请启动终端。然后在终端中输入适当的pip3 install命令。要安装CPU-only版本的TensorFlow,请输入下面的命令:

C:\> pip3 install --upgrade tensorflow

要安装GPU版本的TensorFlow,输入下面的命令:

C:\> pip3 install --upgrade tensorflow-gup

5 通过Anaconda安装

Anaconda安装时社区支持,不是官方支持

按照以下步骤在Anaconda环境中安装TensorFlow:

1.按照Anaconda下载网站上的说明下载并安装Anaconda。

2.通过调用下面命令,创建一个名为tensorflow的conda环境:

C:> conda create –n tensorflow

3.通过执行以下命令激活conda环境:

C:> activate tensorflow

4.执行合适的命令在你的conda环境中安装TensorFlow。要安装CPU-only版本的TensorFlow,输入以下命令:

C:> pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow-1.0.1-cp35-cp35m-win_amd64.wh1

要安装GPU版本的TensorFlow,输入以下命令(只是一行):

C:> pip install --ignore-installed --upgrade https://storage.gooleapis.com/tensorflow/windows/gpu/tensorflow_gpu-1.0.1-cp35-cp35m-win_amd64.wh1

6 验证你的安装

通过以下操作来验证你的TensorFlow安装:

1.启动一个终端

2.如果你是通过Anaconda安装,请激活你的Anaconda环境

3.在终端中,调用python:

C:> python

4.在python交互式shell中输入以下短程序:

>>> import tensorflow as tf

hello = tf.constant('Hello, TensorFlow!')

sess = tf.Session()

print(sess.run(hello))

如果Python程序输出以下内容,则安装成功,你可以开始编写TensorFlow程序了。(如果你是TensorFlow新手,请参阅TensorFlow入门)

Windows10下搭建TensorFlow环境

如果系统生成了错误信息而不是问候(不清楚为什么多了一个b'),看下一节。(这里没有翻译下一节,如有需要去官网上看,或自行搜索,一般会成功)。

上一篇:认真分析mmap:是什么 为什么 怎么用


下一篇:UVa 11082 Matrix Decompressing(最大流)