在 CentOS 8 上安装 TensorFlow

TensorFlow 是 Google 构建的开放源代码的机器学习平台。它可以在不同设备上的 CPU 或 GPU 上运行,并且被许多组织使用,包括 Twitter , PayPal , Intel , Lenovo 和 Airbus 。

TensorFlow 可以在系统范围内安装, 也可以安装在Python 虚拟环境中,或者作为 Docker 容器安装,或者与 Anaconda 一起安装。

本教程说明了如何在 CentOS 8 上安装 TensorFlow 。

TensorFlow 同时支持 Python 2 和3。我们将使用 Python 3 并在虚拟环境中安装 TensorFlow 。虚拟环境使您可以在一台计算机上拥有多个不同的隔离 Python 环境,并在每个项目的基础上安装模块的特定版本,而不必担心这会影响其他项目。

在 CentOS 上安装 TensorFlow

与其他 Linux 发行版不同, CentOS 8 上默认未安装 Python 。要在 CentOS 8 上安装 Python 3 ,请以 root 或 sudo 用户身份在终端中运行以下命令:

sudo dnf install python3

上面的命令将安装 Python 3.6 和 pip 。要运行 Python 3 ,您需要显式键入 python3,同样要运行 pip 需要显示键入 pip3

从 Python 3.6 开始,创建虚拟环境的推荐方法是使用 venv 模块。

导航至您想要存储 TensorFlow 项目的目录。它可以是您的主目录,也可以是用户具有读写权限的任何其他目录。

为 TensorFlow 项目创建一个新目录并切换到该目录:

mkdir tensorflow_project 
cd tensorflow_project

在目录中,运行以下命令以创建虚拟环境:

python3 -m venv venv

上面的命令创建一个名为的目录 venv ,其中包含 Python 二进制文件的拷贝, 标准 Python 库以及其他支持文件。您可以为虚拟环境使用任何名称。

要开始使用虚拟环境,请通过键入以下内容将其激活:

source venv/bin/activate

激活后,虚拟环境的 bin 目录将添加到 $PATH  环境变量的开头。另外,您的 Shell 提示符将更改,并显示您当前正在使用的虚拟环境的名称。在这个例子中为 venv

TensorFlow 安装需要 pip 19 或更高版本。运行以下命令以升级 pip 到最新版本:

pip install --upgrade pip

现在已经创建并激活了虚拟环境,使用以下命令安装 TensorFlow 库:

pip install --upgrade tensorflow

如果您拥有专用的 NVIDIA GPU ,并项利用 NVIDIA GPU 的优势, 请在上面的安装命令中用 tensorflow-gpu 替换 tensorflow ,这包含了 GPU 支持的软件包。

在虚拟环境中,您可以使用命令 pip 代替 pip3, 使用 python 代替 python3

要验证安装,请运行以下命令,该命令将打印 TensorFlow 版本:

python -c 'import tensorflow as tf; print(tf.__version__)'

在撰写本文时, TensorFlow 的最新稳定版本是 2.1.0 :

2.1.0

您的 TensorFlow 版本可能与此处显示的版本不同。

如果您是 TensorFlow 新手,请访问 TensorFlow 入门页面并了解如何构建第一个 ML 应用程序。您还可以从 Github  克隆 TensorFlow 模型TensorFlow-Examples 项目库,并探索和测试 TensorFlow 示例。

完成工作后,通过键入 deactivate 来禁用虚拟环境,然后将返回到常规 shell 。

deactivate

现在,您已经成功安装了 TensorFlow ,就可以开始使用它了。

结论

我们已经向您展示了如何在 CentOS 8 的虚拟环境中安装 TensorFlow 。