原文地址:https://github.com/tensorflow/tensorflow/blob/master/tensorflow/docs_src/install/install_windows.md
在Windows上安装TensorFlow
这篇文章将引导你在Windows上安装TensorFlow。
决定要安装哪个版本的TensorFlow
你需要选择下列类型中的一款TensorFlow来进行安装:
1.只有CPU支持的TensorFlow。如果你的系统没有NVIDIA@GPU,你需要安装这个版本。值得注意的是这个版本的TensorFlow往往安装起来比较简单(一般只需要5到10分钟 ),所以即使你有NVIDIA GPU,我们仍然推荐你先安装这个版本。
2.有GPU支持的TensorFlow。TensorFlow程序在GPU上运行通常比CPU上要快很多。所以如果你的系统有满足以下要求的NVIDIA GPU并且你需要运行性能要求较高的应用,你应该选择这个版本。
运行GPU支持的TensorFlow的要求
如果您使用本指南介绍的机制之一来安装具有GPU支持的TensorFlow,则系统中必须安装以下NVIDIA软件:
1.CUDA® Toolkit 8.0。详细信息请看http://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/。请确保你将关联Cuda的路径名称加入在文档中描述的%PATH%环境变量中。
2.The NVIDIA drivers associated with CUDA Toolkit 8.0
3.cuDNN v6.1。详细信息请看https://developer.nvidia.com/cudnn。请注意,cuDNN通常安装在与其他CUDA DLL不同的位置。 确保将您安装cuDNN DLL的目录添加到%PATH%环境变量中。
4.拥有CUDA Compute Capability 3.0或更高的GPU卡。此列表https://developer.nvidia.com/cuda-gpus中列出了所有支持的GPU。
如果您有其他版本的上述软件包,请更改为指定的版本。 特别地,cuDNN版本必须完全匹配:如果找不到cuDNN64_6.dll,TensorFlow将不会加载。 要使用不同版本的cuDNN,您必须从源代码编译。
决定如何安装TensorFlow
你需要从下列机制中选择一个来安装TensorFlow:
1.本地 pip
2.Anaconda
本地pip直接在您的系统上安装TensorFlow,而无需通过虚拟环境。 由于本地pip安装不会单独被隔离,因此pip安装可能会干扰系统上其他基于Python的安装。 但是,如果您了解pip和您的Python环境,本机点安装通常只需要一个命令! 此外,如果您使用本机点安装,用户可以从系统上的任何目录运行TensorFlow程序。
在Anaconda中,您可以使用conda来创建虚拟环境。 但是,在Anaconda中,我们建议使用pip install命令安装TensorFlow,而不是使用conda install命令。
注意:conda包是社区支持的,没有官方支持。 也就是说,TensorFlow团队既不测试也不维护这个conda包。 使用该包将自行承担风险。
使用本地pip安装
安装下列python中的一个:
Python 3.5.x 64-bit from python.org
Python 3.6.x 64-bit from python.org
TensorFlow在Windows上支持Python 3.5.x和3.6.x。 请注意,Python 3附带了pip3软件包管理器,这是您将用于安装TensorFlow的程序。
要安装TensorFlow,请启动终端。 然后在该终端中发出适当的pip3 install命令。 要安装仅限CPU版本的TensorFlow,请输入以下命令:
C:\> pip3 install --upgrade tensorflow
安装有GPU支持的版本:
C:\> pip3 install --upgrade tensorflow-gpu
使用Anaconda安装(个人推荐):
Anaconda安装是社区支持的,并不由官方提供支持。
下列步骤将在Anaconda环境中安装tensorflow:
1.下载安装Anaconda 详情请看https://www.continuum.io/downloads
2.创建conda环境,命名为tensorflow:
C:\> conda create -n tensorflow python=3.5
3.激活conda 环境:
C:\> activate tensorflow
(tensorflow)C:\> # Your prompt should change
4.选择合适的命令来安装TensorFlow(二选一):
CPU版本:
(tensorflow)C:\> pip install --ignore-installed --upgrade tensorflow
GPU版本:
(tensorflow)C:\> pip install --ignore-installed --upgrade tensorflow-gpu
验证你的安装:
开启一个终端 win+r cmd
如果你采用anaconda安装,激活你的环境 activate tensorflow
唤醒python,进入环境
$ python
述如下列命令:
>>> import tensorflow as tf >>> hello = tf.constant('Hello, TensorFlow!') >>> sess = tf.Session() >>> print(sess.run(hello))如果出现下列现象,说明安装成功:
Hello, TensorFlow!至此安装过程全部翻译完成,还有一些常见的安装问题,请具体参看原网址:
https://github.com/tensorflow/tensorflow/blob/master/tensorflow/docs_src/install/install_windows.md