Windows Tensorflow 环境搭建

Windows Tensorflow 环境搭建

Python 安装

推荐安装3.6.3版本，下载以后，一直next即可。
下载地址：
https://www.python.org/ftp/python/3.6.3/python-3.6.3-amd64.exe

安装后的测试方法，cmd，python，显示Python版本号为3.6.3

Python环境变量

右键电脑 > 属性 > 高级系统设置 > 高级 > 环境变量
系统变量 > 新建
PY_HOME
你的Python路径
例如：
PY_HOME D:\test\Python363
添加后，在全局变量后添加以下环境变量，注意用;分割
%PY_HOME%\
%PY_HOME%\Scripts\

配置script环境变量，主要用来调用pip等Python内置脚本或者工具

TensorFlow 安装

使用国内源能有效改善下载速度，此处使用的是豆瓣的源
打开cmd，输入以下命令即可。

pip install -i https://pypi.douban.com/simple/ tensorflow

如果硬件有N卡（NVIDIA显卡）支持，想要安装GPU版本的TensorFlow，命令为

pip install -i https://pypi.douban.com/simple/ tensorflow-gpu

并且要安装N卡驱动以及cuda和cuDNN
下载地址：
cuda：https://developer.nvidia.com/cuda-toolkit-archive
cuDNN：https://developer.nvidia.com/cudnn
注意：
cuDNN的下载需要先注册账号，此处不再赘述。
TensorFlow目前支持cuda的版本为cuda8，不要安装最新的cuda9。
cuDNN的下载也要对应cuda版本。

cuDNN下载完成后，将lib复制到cuda的安装目录下即可。

以上步骤完成后，TensorFlow环境就搭建完成了，很简单吧。
测试一下：
cmd

Python
import tensorflow as tf

没有报错，证明TensorFlow CPU环境搭建成功

Python
import tensorflow-gpu as tf

没有报错，证明TensorFlow GPU环境搭建成功