如何在Ubuntu 20.04上安装TensorFlow

TensorFlow是Google建立的一个免费的开源机器学习平台。 许多组织(包括Twitter,PayPal,英特尔,联想和空中客车公司)都使用它。

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

本教程说明了如何在Ubuntu 20.04的Python虚拟环境中安装TensorFlow。

虚拟环境允许您在一台计算机上拥有多个不同的隔离Python环境,并在每个项目的基础上安装模块的特定版本,而不必担心会影响您的其他Project。

在Ubuntu 20.04上安装TensorFlow#

默认情况下,Ubuntu 20.04随附Python 3.8。 您可以通过键入以下命令来验证系统上已安装Python 3:

python3 -V

输出应如下所示:

Python 3.8.2

建议的创建虚拟环境的方法是使用 venv 模块,该模块包含在 python3-venv 包。

要安装 python3-venv 程序包运行以下命令:

sudo apt install python3-venv python3-dev

一旦安装了模块,就可以为TensorFlow项目创建虚拟环境了。

导航到您要存储Python 3虚拟环境的目录。 它可以是您的主目录,也可以是用户具有读写权限的任何其他目录。

为TensorFlow项目创建一个新目录并cd到其中:

mkdir my_tensorflowcd my_tensorflow

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

python3 -m venv venv

第二 venv 是虚拟环境的名称。 您可以为虚拟环境使用任何名称。

上面的命令创建一个名为 venv,其中包含Python二进制文件,Pip程序包管理器,标准Python库和其他支持文件的副本。

要开始使用虚拟环境,请通过运行 activate 脚本:

source venv/bin/activate

激活后,虚拟环境的bin目录将添加到系统的开头 $PATH 可变的。 而且,shell的提示符将发生变化,并且将显示您当前所在的虚拟环境的名称。在此示例中,即 (venv)

TensorFlow安装要求 pip 版本19或更高版本。 输入以下命令进行升级 pip 到最新版本:

pip install --upgrade pip

现在已经激活了虚拟环境,是时候安装TensorFlow软件包了。

pip install --upgrade tensorflow

如果您拥有专用的NVIDIA GPU,并希望利用其处理能力,而不是 tensorflow,安装 tensorflow-gpu 软件包,其中包括GPU支持。

在虚拟环境中,可以使用以下命令 pip 代替 pip3python 代替 python3

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

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

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

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

2.2.0

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

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

完成工作后,请输入以下内容停用环境 deactivate,您将返回到常规shell。

deactivate

结论#

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

如果您遇到问题或有反馈,请在下面发表评论。

python ubuntu

Sidebar