频道栏目
首页 > 网络 > 云计算 > 正文

Windows下安装tensorflow步骤

2016-12-29 11:05:00         来源:gdyshi的专栏  
收藏   我要投稿

官方安装链接
tensorflow 0.12.0开始支持Windows下安装了

环境要求:

Windows 64位 python 3.5 pip 9.0.1 tensorflow 0.12.0 cuda8.0 cudnn5.1
查看gpu是否支持cuda

安装过程:

安装python3.5

下载 安装 升级pip
python.exe -m pip install --upgrade pip

安装tensorflow

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

gpu版本tensorflow安装gpu支持包

安装cuda8.0

下载 安装

安装cudnn5.1

下载 安装
添加解压后的bin目录到PATH环境变量

测试tensorflow

$ python
...
>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print(sess.run(hello))
Hello, TensorFlow!
>>> a = tf.constant(10)
>>> b = tf.constant(32)
>>> print(sess.run(a + b))
42
>>>
gpu使用情况查看
nvidia-smi -l # 实时返回GPU使用情况
上一篇:再部署一个instance和LocalNetwork-每天5分钟玩转OpenStack(131)
下一篇:OpenStack网络指南(12)BGP动态路由
相关文章
图文推荐

关于我们 | 联系我们 | 广告服务 | 投资合作 | 版权申明 | 在线帮助 | 网站地图 | 作品发布 | Vip技术培训 | 举报中心

版权所有: 红黑联盟--致力于做实用的IT技术学习网站