![TensorFlow深度学习从入门到进阶](https://wfqqreader-1252317822.image.myqcloud.com/cover/944/36922944/b_36922944.jpg)
1.6 搭建环境
本节主要介绍在Windows的平台上如何安装TensorFlow,以及简单的运行测试。
1.6.1 安装环境
因为深度学习计算过程中大量的操作是向量和矩阵的计算,而GPU在向量和矩阵计算方面比CPU有一个数量级的速度提升,所以机器学习在GPU上运算效率更高。
通过以下方式来查看Windows系统上的GPU信息。
在“运行”对话框中输入dxdiag,如图1-2所示,然后单击“确定”按钮,此时会打开“DirectX诊断工具”对话框。单击其中的“显示”选项卡,可以查看机器的显卡信息,如图1-3所示。
![](https://epubservercos.yuewen.com/C16494/19573973401151406/epubprivate/OEBPS/Images/978-7-111-65263-2_16_01.jpg?sign=1738977155-fAW2zVGixxYceqUVmnKIZJow6uR8QSHT-0-5f61dbb4c2281d4be78a3b6a7bd10514)
图1-2 输入dxdiag命令
![](https://epubservercos.yuewen.com/C16494/19573973401151406/epubprivate/OEBPS/Images/978-7-111-65263-2_16_02.jpg?sign=1738977155-Q2IwpXhqWPLCwfMkO5rWdGvKWpkbe01a-0-5842b5ff870aa46cb4e2ce764797e6ae)
图1-3 查看Windows的显卡信息
由图1-3可以看到,这个机器上的显卡芯片类型是Intel(R) HD Graphics Family。
1.6.2 安装TensorFlow
TensorFlow的Python语言API支持Python 2.7和Python 3.3以上的版本。本书使用的是TensorFlow 3.6.5版本。
1.安装pip
pip是用来安装和管理Python包的管理工具。首先,去Python官网下载pip最新版本(https://pypi.python.org/pypi/pip#downloads),下载完成后,在Windows系统上安装pip的命令为:
python setup.py install
接着在Windows中设置环境变量,方法为在Windows环境变量的PATH变量后添加“\Python安装目录\Scripts”。
目前,TensorFlow在Windows上只支持64位的Python 3.6.5版本。
2.通过pip安装TensorFlow
TensorFlow已经把最新版本的安装程序上传到了Pypi,所以可以通过最简单的方式来安装TensorFlow(要求pip版本在8.1版本或者更高)。
安装CPU版本的TensorFlow的命令如下:
![](https://epubservercos.yuewen.com/C16494/19573973401151406/epubprivate/OEBPS/Images/978-7-111-65263-2_17_01.jpg?sign=1738977155-MCdDoRbrJpJfsq3EOtpjEZ5ZLNBrPHh7-0-2769f3b66451ba0fd1b1d94736683ee7)
安装支持GPU版本的TensorFlow的命令如下:
![](https://epubservercos.yuewen.com/C16494/19573973401151406/epubprivate/OEBPS/Images/978-7-111-65263-2_17_02.jpg?sign=1738977155-ODf0TvHlpYUi0cLwzt1Po47m2MDvt0rP-0-7bb52ba28651bd7e8b70e5c9019f1b48)
在Windows系统上安装CPU版本(0.12版本)的命令如下:
![](https://epubservercos.yuewen.com/C16494/19573973401151406/epubprivate/OEBPS/Images/978-7-111-65263-2_17_03.jpg?sign=1738977155-UnRz3rAAVMBnbyfj3vSroT1DVJNWIRX9-0-cde747464092404e2ef8a3d2330fa852)
TensorFlow在Windows上依赖MSVCP140.DLL,这里需要提前安装Visual C++2015 redistributable(x64位),其下载地址为https://www.microsoft.com/en-us/download/details.aspx?id=53587,下载文件为vc_redist.x64.exe。
1.6.3 安装测试
如果顺利的话,到这里已经成功安装了TensorFlow,那么简单测试一下安装是否成功。
![](https://epubservercos.yuewen.com/C16494/19573973401151406/epubprivate/OEBPS/Images/978-7-111-65263-2_17_04.jpg?sign=1738977155-MXRJy7wXC94HXyYocfDJKNOGofJl7QIJ-0-0130af549fc4baa4e88e6c253570eb40)
上面这段代码若正常运行,会打印出TensorFlow的版本号,这里是“1.7.0”。
但也可能会存在一些问题:
如果在import tensorflow as tf之后,打印出来Cuda的so或者CuDNN的so没有找到,一般是因为Cuda或者CuDNN的路径没有添加到环境变量里。
下面再进行一个简单的计算,看看TensorFlow是否运行正常。输入如下代码:
![](https://epubservercos.yuewen.com/C16494/19573973401151406/epubprivate/OEBPS/Images/978-7-111-65263-2_18_01.jpg?sign=1738977155-aDJtWsyJMgaIPvk5iwit0qAyYaA6pUeP-0-d54c27d4be1cf4536ac03a54bcdb379a)
如果这段代码可以正常输出“Hello,TensorFlow!”和“2+3=5”,那么说明TensorFlow已经成功安装了。