在 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 。