今天我们聊聊如何在windows操作系统环境下,安装TensorFlow2.X版本。今天的话题分为三个部分:
(1)硬件环境配置;
(2)软件选择;
(3)安装及问题处理。
![efa08b4cdf9e0319434baa01b018358b.png](https://i-blog.csdnimg.cn/blog_migrate/022e893faaa7a3fe5861c201ca929e44.jpeg)
(1)硬件环境配置
通常,在开发环境下,安装TensorFlow2.X的硬件配置主要包括以下几项。
- CPU、GPU、TPU(张量处理单元)
- 内存
- 硬盘
CPU作为基础配置项,有经济条件,最好选择最新一代的Intel多核CPU。对于比较老的CPU或者是ARM,在安装过程中,可能会出现TF2.X不支持的情况。留意安装过程中的提示信息,如果不幸中枪,那就考虑升级硬件。如果有更多的经济资源,可以选择配置GPU,甚至TPU(张量处理单元)。
内存当然是越大越好,最基本的要求是内存容量大于16GB。
硬盘主要是IO和容量空间,建议选择SSD类型,至少要留有100GB的硬盘空间,避免由于空间不够,导致系统无法安装和运行过程中,出现磁盘没有剩余空间的尴尬。
本次安装是基于ThinkPad笔记本,Inteli5+16GB内存+150GB硬盘空间,没有GPU和TPU。
(2)软件选择。
以下操作系统(64 位系统)和软件是TensorFlow官方已测试通过的,TF2.X支持的软件环境。
- 操作系统选择
- Ubuntu 16.04 或更高版本
- Windows 7 或更高版本(含 C++ 可再发行软件包)
- macOS 10.12.6 (Sierra) 或更高版本(不支持 GPU)
- Raspbian 9.0 或更高版本
- Python版本
- Python 3.5–3.8
除了上面提到的TF2.X支持的Ubuntu、window7、MacOS和Rasbian外,Centos应该是也是可以支持的。
在学习环境下,我们采用比较普遍的Win7(win10类似)+Python3.7。另外,关于是否要先安装Anaconda的问题。看学员自己的技能水平。Anaconda是管理Python library的工具。对于比较熟练Python和操作系统的学员和用户,可以不安装这个工具。如果不是很熟练的学员。建议先安装Anaconda。
- 安装TensorFlow2.X
在第一步和第二步都正常完成的情况下。把TensorFlow2.X当做Patho的一个library,在window CMD(命令行环境)下,键入如下命令,即可顺利完成TF2.X的安装。
pip install tensorflow -i https://pypi.tuna.tsinghua.edu.cn/simple #选择国内源安装,速度快。
如果已安装,需要更新Tensorflow版本,可键入:
pip install --upgrade tensorflow -i https://pypi.tuna.tsinghua.edu.cn/simple
如果以上命令执行完毕,没有出现错误提示,表明已经顺利完成TensorFLow2.X的安装。下面我们测试一下安装结果。
启动python,导入tensorflow,打印tensorflow的版本号。
![d5107e7e8c4effb8d6f948cc10601442.png](https://i-blog.csdnimg.cn/blog_migrate/8a80bde75215f1b2358cb5fc2209614d.jpeg)
测试TensorFlow安装结果,打印TF版本信息
说明:
- TensorFlow 的最新安装步骤可参考官方网站上的说明(https://tensorflow.google.cn/install)
- 从 TensorFlow 2.1 开始,采用pip命令工具安装 tensorflow的同时包含 GPU 支持,无需单独安装Tensorflow GPU 版本。但如果对 pip 包的大小敏感,可使用 tensorflow-cpu 包安装仅支持 CPU 的 TensorFlow 版本。
- 在键入pip命令之前,最好先键入以下命令,升级pip命令工具的版本: python -m pip install --upgrade pip
- python运行环境下,检测GPU是否可用的语句: tf.test.is_gpu_available()
成功的方法只有一个,不成功的原因各有不同。如果在安装过程出现了错误提示。需要仔细查找原因,并修正后,才可以执行下一步。
这里讲解碰到的一个比较典型的问题及解决方法。
错误提示信息(贴出重点关注的信息,其他已省略):
...................Could not load dynamic library ‘cudart64_101.dll’; dlerror: cudart64_101.dll not found..................
这个提示信息的含义是:“”无法加载动态库cudart64_101.dll;出现报错:未找到cudart64_101.dll。“
解决方法:
(1)下载cudart64_101.dll文件 (下载地址链接: https://www.dll-files.com/cudart64_101.dll.html)
(2)将解压后的dll文件放在:
![0e6463a3fd15ddadff50cd17a5abb746.png](https://i-blog.csdnimg.cn/blog_migrate/6dfb1d971ea5c0081a0656b032e8b5b3.jpeg)
(3)再次测试,出现以下信息,表明问题已解决。
“Successfully opened dynamic library cudart64_101.dll”
今天就到这里。