Windows下TensorFlow安装与代码测试
Windows下TensorFlow安装与代码测试
一:Tensorflow介绍
TensorFlow是谷歌的深度学习应用开发框架,其思想基于数据流图与节点图实现,简单易用,同时有具有以下几个特点:
可视化训练过程
CPU与GPU支持
先构建再运行的数据流节点会话
支持的系统包括如下:
Ubuntu 16.04以后版本
Windows7以后版本
MacOS X 10.11以后版本
特别强调一下,最好是64位的系统比较好,因为Tensorflow本身测试都是64位的机器上完成的,当前的Tensorflow的最新版本为1.6。除了Tensorflow是深度学习的开发框架,其它常见的框架还包括
DeepLearning4J
Torch
Keras
Caffe
CNTK
Theano
此外,国内的百度,腾讯都开源了自己的深度学习框架,感兴趣的可以了解一下,当前使用人数最的应该是Tensorflow与Caffe。GITHUB上的统计数据也说明了这一点, 看下图:
很明显TF在各个指数上都是最高的,说他是最流行的深度学习开发框架一点也没错。
二:Tensorflow安装
以下的操作适合Windows7与Windows10 64位系统,Tensorflow官方支持的语言最主要的是Python,其它还包括Java、Go、C++等。这里以Python为例, 安装Tensorflow1.6首先需要安装Python3.6,然后可以下载安装PyCharm-IDE开发工具,如果一切正常,这个时候打开你的命令行工具输入如下命令
python -m pip install --upgrade tensorflow
python -m pip install --upgrade tensorflow-gpu // 支持GPU
如果上述命令不工作,还可以尝试下面两个
pip3 install --upgrade tensorflow
pip3 install --upgrade tensorflow-gpu // 支持GPU
上述的命令行,其中GPU支持的只有在机器有GPU支持的显卡上才需要执行支持GPU。另外你可能还需要执行一些其他的pip指令安装numpy、scikit-learn、scipy等。
三:代码演示
如果一切OK,这个时候你启动Python的IDE-Pycharm就会看到这样的界面
新建一个python文件,在里面写下如下代码:
import tensorflow as tf
import numpy as np
with tf.Session() as sess:
a = np.array([[2, 3]], dtype=np.float32)
b = np.array([[3], [2]], dtype=np.float32)
print("a=", a)
print("b=", b)
c = tf.matmul(a, b)
result = sess.run(c)
print("result",result)
执行得到如下结果:
a= [[ 2. 3.]]
b= [[ 3.]
[ 2.]]
result [[ 12.]]
上述代码实现了两个矩阵的乘法运算,如果得到上面的输出结果,说明你的tensorflow已经成功安装,可以开始在Windows系统上开始使用了。
注意:
如果有GPU支持,需要安装CUDA开发工具包与cuDNN,tensorflow1.6支持的版本是9.0。这个可以去英伟达的网站上下载即可。
远飞者当换其新羽
更多精彩内容请关注:
续更多tensorflow相关文章,请持续关注!