终于迈开了这一步,TensorFlow 大法好。

一、介绍

TensorFlow™ 是一个采用数据流图(data flow graphs),用于数值计算的开源软件库。节点(Nodes)在图中表示数学操作,图中的线(edges)则表示在节点间相互联系的***数据数组,即张量(tensor)。它灵活的架构让你可以在多种平台上展开计算,例如台式计算机中的一个或多个 CPU(或 GPU),服务器,移动设备等等。TensorFlow 最初由 Google 大脑小组(隶属于 Google 机器智能研究机构)的研究员和工程师们开发出来,用于机器学习和深度神经网络方面的研究,但这个系统的通用性使其也可广泛用于其他计算领域。(来自TensorFlow中文社区

一开始还没发现这图标的意思,直到偶然间,发现正好是 TF 的影子重叠在一起构成的。TensorFlowLOGO 是真的好看。

二、安装 TensorFlow

我是直接用 pip 安装的 GPU 版本。

pip3 install --upgrade tensorflow-gpu

这个安装好之后,还不能跑 TensorFlow ,还缺两个东西。

三、安装 CUDA

这个你要确定版本,非常重要!否则不能使用。可以先找到对应的显卡:网址在这里。我的是 GTX960M 的显卡。

我当时按照上面的指令安装 TensorFlow 的时候,自动装的是比较新的版本。然后我就去装了最新的 CUDA ,结果发现不能用,又试了几个版本,最终装了 CUDA 9.0 才成功。

先去官网找到对应的版本。

再选择相应的信息,点击 Download 即可。

下载完后,直接点击 exe 进行安装。下面演示的是 CUDA 10 安装过程。


选择 精简 安装。

四、安装 cuDNN

首先,要去官网注册个账号。然后,选择与 CUDA 相对应的版本。

下载完解压,我是把 cuda 文件夹解压到 C盘 的。

添加环境变量。

五、测试

import tensorflow as tf
a = tf.random_normal((100, 100))
b = tf.random_normal((100, 500))
c = tf.matmul(a, b)
sess = tf.InteractiveSession()
sess.run(c)

如果看到运行出下图的结果,就说明安装成功了。

六、致谢

【TensorFlow】Windows10 64 位下安装 TensorFlow - 官方原生支持

win10 + gtx1060 + cuda8.0 + Tensorflow gpu 版本安装

Tensorflow 安装在 windows 上面出现 ImportError: DLL load failed: 动态链接库 (DLL) 初始化例程失败

Windows 下安装 TensorFlow 快速教程

[window10 上安装 python+CUDA+CuDNN+TensorFlow](https://www.cnblogs.com/touch-skyer/p/8367706.html)