![TensorFlow 2.0卷积神经网络实战](https://wfqqreader-1252317822.image.myqcloud.com/cover/133/29977133/b_29977133.jpg)
1.2 TensorFlow 2.0 GPU版本的安装
Python运行环境调试完毕后,下面的重点就是安装本书的主角TensorFlow 2.0。
1.2.1 检测Anaconda中的TensorFlow版本
首先是对于版本的选择,读者可以直接在Anaconda命令端输入一个错误的命令:
![](https://epubservercos.yuewen.com/5F793F/16499866704817006/epubprivate/OEBPS/Images/Figure-P18_26965.jpg?sign=1739260233-pzijoiUwfxB7Ea8EdVSaha2TcmqlOfAX-0-93514e81a7417c6ead4fa9f72e7a0a44)
这个命令是错误的,目的是为了查询当前的TensorFlow版本,作者在写作这本书所能获取的TensorFlow版本如图1.16所示。
![](https://epubservercos.yuewen.com/5F793F/16499866704817006/epubprivate/OEBPS/Images/Figure-P19_3840.jpg?sign=1739260233-73dvqHrDRSWBJTcUPw95IbwtOpoBurdt-0-48e83eb8f9166c8cd1f555c4a052a88b)
图1.16 TensorFlow版本汇总
可以看到,目前最新的版本是2.0.0b1。此时,如果读者想安装CPU版本的TensorFlow,直接在当前的Anaconda输入命令如下:
![](https://epubservercos.yuewen.com/5F793F/16499866704817006/epubprivate/OEBPS/Images/Figure-P19_26967.jpg?sign=1739260233-slbUrTLMLII0BsO3RM0A8RfUc3crga08-0-825d3229a93aeb06c291a8efadc7fdf3)
即可安装最新CPU版本的TensorFlow。
1.2.2 TensorFlow 2.0 GPU版本基础显卡推荐和前置软件安装
如果从CPU版本的TensorFlow 2.0开始你的深度学习之旅,这是完全可以的。但是却不是作者推荐的一种方式。相对于GPU版本的TensorFlow来说,其运行速度CPU版本存在着极大的劣势,很有可能会让你的深度学习止步于前。
实际上,配置一块TensorFlow 2.0 GPU版本的显卡(见图1.17)并不需要花费很多,从网上购买一块标准的NVIDA 750ti显卡就能够基本满足读者起步阶段的基本需求,作者在这里强调的是,最好购置显存为4G的版本,目前价格稳定在400元左右。如果有更好的条件的话,NVIDA 1050ti 4G版本也是一个不错的选择,价格在700元左右。
![](https://epubservercos.yuewen.com/5F793F/16499866704817006/epubprivate/OEBPS/Images/Figure-P19_3878.jpg?sign=1739260233-vPvVYfT9FsvoDyWGeLgDkYmA5V3u6t1c-0-a064fe450130a683dc49617d445bedf2)
图1.17 深度学习显卡
注意
推荐购买NVIDA系列的显卡,并且优先考虑大显存的。
下面就介绍TensorFlow 2.0 GPU版本的前置软件的安装。对于GPU版本的TensorFlow 2.0来说,由于调用了NVIDA显卡作为其代码运行的主要工具,因此额外需要NVIDA提供的运行库作为运行基础。
(1)首先介绍版本的问题,作者目前使用的TensorFlow 2.0运行的NVIDA运行库版本如下:
●CUDA版本:10.0。
●CuDNN版本:7.5.0。
这个对应的版本一定要配合使用,建议读者不要改动,直接下载对应版本就可以。
CUDA的下载地址为:https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal。界面如图1.18所示。
直接下载local版本安装即可。
![](https://epubservercos.yuewen.com/5F793F/16499866704817006/epubprivate/OEBPS/Images/Figure-P20_3937.jpg?sign=1739260233-FUWR4U2dH3cBxlwLFmXw34bbzi42Rrkp-0-4c151e1bb56663c5f18113ab581e85cc)
图1.18 下载CUDA文件
(2)下载后是一个exe文件,读者自行安装即可,不要修改其中的路径信息,完全使用默认路径安装即可。
(3)接着是下载和安装对应的cuDNN文件。下载地址为:https://developer.nvidia.com/rdp/cudnn-archive。
cuDNN的下载需要先注册一个用户名,相信读者可以很快完成,之后直接进入下载页面,如图1.19所示。
![](https://epubservercos.yuewen.com/5F793F/16499866704817006/epubprivate/OEBPS/Images/Figure-P21_3955.jpg?sign=1739260233-p363DsWsHvYqMxe6LsYLAOy3oB8Bipy2-0-afdb6aa608145f9a1d4ca2f7f033889a)
图1.19 下载cuDNN文件
注意
不要选择错误的版本,一定需要找到对应的版本号。
(4)下面就是cuDNN的安装问题,下载的cuDNN是一个压缩文件,直接将其解压到CUDA安装目录即可,如图1.20所示。
![](https://epubservercos.yuewen.com/5F793F/16499866704817006/epubprivate/OEBPS/Images/Figure-P21_3992.jpg?sign=1739260233-SKCEsJ2BWVTRvZCwY9hSA7RNwcNEnWyr-0-a5b4fec6e48c0c9e4552f12bd33b196f)
图1.20 CUDA安装目录
(5)接下来就是对环境变量的设置,这里需要将CUDA的运行路径加载到环境变量的path路径中,如图1.21所示。
![](https://epubservercos.yuewen.com/5F793F/16499866704817006/epubprivate/OEBPS/Images/Figure-P22_26973.jpg?sign=1739260233-9uxwJPWTyMqmxABq8jRPqFvxekJRYEgU-0-1f4e6583856a0c146f3df8b571c434c2)
图1.21 将CUDA路径加载到环境变量的path中
(6)最后完成TensorFlow 2.0 GPU版本的安装了,只需一行简单的代码:
![](https://epubservercos.yuewen.com/5F793F/16499866704817006/epubprivate/OEBPS/Images/Figure-P22_26974.jpg?sign=1739260233-piWc4FfQlQgWYw44iuDw1LxZooMO13DC-0-a018e033e5e81ec60dc722325591671c)