Win10 TensorFlow1.4 (gpu)安装

TensorFlow 1.4 版现在已公开发布 - 这是一个大更新!这里有一些新功能

Keras

在1.4 版中,Keras 已从tf.contrib.keras 迁移到核心软件包 tf.keras 中。Keras是一个非常热门的机器学习框架,它包含众多高级 API,这些 API 可以最大程度缩短从您的创意到可行实现之间的时间。

Keras 可与其他核心 TensorFlow 功能平稳集成,包括 Estimator API。事实上,您可以调用 tf.keras.estimator.model_to_estimator函数,直接从任何 Keras 模型构建估算器。由于 Keras 现在已添加到 TensorFlow 核心中,您可以在生产工作流程中依赖它。

数据集

Dataset API 已从tf.contrib.data 迁移到核心软件包 tf.data中。1.4 版的 Dataset API 还增加了对 Python 生成器的支持。强烈建议使用 Dataset API 为 TensorFlow 模型创建输入管道,因为:

  • 与旧 API(feed_dict 或队列式管道)相比,Dataset API 可以提供更多功能。
  • Dataset API 的性能更高。
  • Dataset API 更简洁,更易于使用。
  •  

其他增强功能

·        除了这里介绍的功能外,1.4 版还引入了许多其他增强功能,版本说明中进行了相关介绍:

·        https://github.com/tensorflow/tensorflow/blob/master/RELEASE.md

之前,我tensorflow1.2用的好好的,手贱升级了,才知道cuda,cudnn的版本以及不支持了,所以今天重新安装了。

配置要求

具体CUDA和CUDNN的版本要求取决于你要装的tf的版本要求,1.4至少要CUDA8.0 + CUDNN V6.1,故此我安装的配置就是: 
python 3.6 + CUDA 8.0 + CUDNN V6.1 + tf 1.4

Win10 TensorFlow1.4 (gpu)安装

Tensorflow 官网被墙了 2333


python安装

       先下载python 3.6,我强烈推荐暗转anaconda,十分强大的python科学计算平台,这样会省去很多麻烦。

https://www.anaconda.com/download/

Win10 TensorFlow1.4 (gpu)安装

Win10 TensorFlow1.4 (gpu)安装


TensorFlow安装

输入:pip install tensorflow-gpu 安装,如果提示pip版本问题,按提示更新到最新版本即可。

 

需要GPU加速,还需要安装cudacuDnn(专门为deep learning准备的加速库)

 

cuda安装

cuda v8.0安装包

Win10 TensorFlow1.4 (gpu)安装

下载下来,双击就可以开始安装了,照着屏幕提示就可以了



CUDNN下载

官网:https://developer.nvidia.com/cudnn要先注册



Win10 TensorFlow1.4 (gpu)安装


下载下来解压,内部目录是这样的

Win10 TensorFlow1.4 (gpu)安装


就是3个目录binincludelib,全部拷贝到你CUDA的目录下就行了,CUDA的目录默认在C:\Program Files\NVIDIA GPU ComputingToolkit\CUDA\v8.0


官方教程代码测试:

import tensorflow as tf

with tf.device('/cpu:0'):

  a =tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')

  b =tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')

c = tf.matmul(a, b)

# 新建session withlog_device_placement并设置为True.

sess =tf.Session(config=tf.ConfigProto(log_device_placement=True))

# 运行这个op.

print (sess.run(c))

Win10 TensorFlow1.4 (gpu)安装

成功了,又可以愉快的撸 tensorflow了