![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=1739251193-7sw7Vk7Xel0y7BKDf1HqA2MyhbDqKOEh-0-47cab2bf423335c6b7391967374a2524)
图1-2 输入dxdiag命令
![](https://epubservercos.yuewen.com/C16494/19573973401151406/epubprivate/OEBPS/Images/978-7-111-65263-2_16_02.jpg?sign=1739251193-ETaLu0K0JoLjZeWlUB39ExT8ej2ZvOz9-0-c7616318090c53b9b40d6037d9cb294d)
图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=1739251193-TwbpsTHWzHhsdh6DmZU8zcqEAvY2v3d0-0-d394b603c3c8bc64e01ddac7bfe233d9)
安装支持GPU版本的TensorFlow的命令如下:
![](https://epubservercos.yuewen.com/C16494/19573973401151406/epubprivate/OEBPS/Images/978-7-111-65263-2_17_02.jpg?sign=1739251193-cRIwPB1xvZLy2rul9bUyf8NNXoNpr0mo-0-f7a81a20b8c3c91a2bf7b5708c4c455f)
在Windows系统上安装CPU版本(0.12版本)的命令如下:
![](https://epubservercos.yuewen.com/C16494/19573973401151406/epubprivate/OEBPS/Images/978-7-111-65263-2_17_03.jpg?sign=1739251193-vzGA29522xJW6cSju9FgNNMeL7tlJTfp-0-09f00ccac2dcfd888fc7c1b209be482a)
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=1739251193-U1uqXherZasbynrcmIzt7zP1L6NVG2Uz-0-9214272b6159fe92e85f83c13b35541c)
上面这段代码若正常运行,会打印出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=1739251193-zdubhuQIDwTigiF4qBUKAa7sDDiStZEO-0-dccb475516b185a1a1ebcb68f70b954d)
如果这段代码可以正常输出“Hello,TensorFlow!”和“2+3=5”,那么说明TensorFlow已经成功安装了。