TensorFlow on Windows10

字数 322阅读 112

安装工具

在Power Shell下执行,“开始”->“附件”->“Windows Power Shell”->“Windows Power Shell”

Python

-安装Anaconda 4.2.0 For Windows
Anaconda 4.2.0 For Windows
-设置环境变量
将Python安装路径下“%安装路径%\Scripts”添加到Path下;再到PowerShell中输入pip看到若干命令提示,则代表python安装成功(Python安装包自带pip)。“开始”->“所有程序”,也可以找到Python终端。

安装CUDA

CUDA安装:CUDA下载

安装CUDNN

CuDNN安装:CUDNN下载
(要注册Nvidia用户,并加入CuDNN开发组,填若干问卷就可以下载了)
选择下载版本时要注意和Cuda版本匹配。解压后覆盖至CUDA的安装目录下,例如:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\

修改国内Pip源

为了使用国内镜像加速pip安装,需要如下修改:
WIndows 7 在“C:\Users\用户名\AppData\Local\pip”文件夹下,新建文本文件,添加内容:

[global]  
index-url = http://mirrors.aliyun.com/pypi/simple/  
[install]  
trusted-host=mirrors.aliyun.com  

关于修改pip国内源可参考这里

安装TensorFlow

Power Shell下输入:

pip install --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-0.12.0rc0-cp35-cp35m-win_amd64.whl  

测试TensorFlow

>>>import tensorflow as tf  
>>>sess = tf.Session()  
>>>a = tf.constant(10)  
>>>b = tf.constant(22)  
>>>print(sess.run(a + b))  
32  

正确输出结果32,安装完毕。

推荐阅读更多精彩内容