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
Tensorflow 官网被墙了 2333
python安装
先下载python 3.6,我强烈推荐暗转anaconda,十分强大的python科学计算平台,这样会省去很多麻烦。
https://www.anaconda.com/download/
TensorFlow安装
输入:pip install tensorflow-gpu 安装,如果提示pip版本问题,按提示更新到最新版本即可。
需要GPU加速,还需要安装cuda和cuDnn(专门为deep learning准备的加速库)。
cuda安装
cuda v8.0安装包
下载下来,双击就可以开始安装了,照着屏幕提示就可以了
CUDNN下载
官网:https://developer.nvidia.com/cudnn要先注册
下载下来解压,内部目录是这样的:
就是3个目录bin、include、lib,全部拷贝到你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))
成功了,又可以愉快的撸 tensorflow了