[TensorFlow] win10安装TensorFlow

安装指南

  Instaling TensorFlow on Windows

安装CUDA Toolkit 8.0

  CUDA Toolkit安装说明
  CUDA Toolkit下载地址
  选择对应的Toolkit下载，在win10下，双击.exe安装即可。安装完毕后，需要看一下环境变量是否配置。

图1 下载CUDA Toolkit

安装cuDNN v5.1

  cuDNN下载地址
  下载后得到的是.h, .dll, .lib三个文件。可以把bin, include, lib三个目录一起放在一个单独的目录下，然后将该路径添加到环境变量中。也可以直接将这三个文件分别放到CUDA Toolkit目录的bin, include, lib目录下。CUDA Toolkit默认的安装路径为C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0。

安装Anaconda

  我是通过Anaconda安装TensorFlow的。
  Anaconda下载地址
  到这里下载Anaconda并安装即可，我用的是4.3.8, 64bit。

Anaconda中创建虚拟环境

  可以在命令行中创建，也可以通过Anaconda Navigate创建。

 命令行创建

  打开windows的命令行。

conda create -n tensorflow python=3.5

图2 命令行安装虚拟环境

注意加上python=3.5，这个TensorFlow的指南里边没讲。TensorFlow需要python3.5，但Anaconda默认安装的是python3.6，因此需要指定python版本。
这里我创建的虚拟环境名字为py35，因为我之前已经创建过一个叫tensorflow的虚拟环境了。

 Anaconda Navigate创建

  Navigate创建虚拟环境也很方便。打开Anaconda Navigate，面板左边选择Environment，然后下面有个create。然后输入环境的名字，选择python版本即可。

图3 Navigate创建虚拟环境

进入虚拟环境安装tensorflow

  做好了前面所有的准备工作，我们开始正式安装tensorflow了。在命令行里进入上一步创建的虚拟环境，通过pip安装tensorflow。

activate tensorflow
pip install –ignore-installed –upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-1.0.1-cp35-cp35m-win_amd64.whl

 

这里大概需要能访问到tensorflow所在的链接。

测试安装

  完成了上一步之后，我们在虚拟环境中就已经安装好tensorflow了。接下来我们通过一个示例程序测试一下tensorflow是否安装正确。在命令行中进入python，执行hello程序。

python
>>> import tensorflow as tf
   he = tf.constant(‘Hello, TensorFlow!’)
   sess = tf.Session()
   print(sess.run(hello))

  如果最终得到字符串Hello, TensorFlow!，表示tensorflow已经正确安装。

图4 Hello, TensorFlow!

这里我还暂时不清楚为什么输出了这么多调试信息。感觉可能有点问题>_<