在 Ubuntu 上安装 TensorFlow 的指南

如果你喜欢机器学习,你可能需要利用 TensorFlow, 如果不 PyTorch.

假设您使用的是 Linux,我将向您展示如何在 Ubuntu 上安装 Tensorflow。 我使用的是 Ubuntu 18.04,但这些步骤也应该适用于其他版本。

请注意,虽然您可以尝试 从源头构建它 对于其他平台/发行版,TensorFlow 主要支持 Ubuntu Linux。 因此,我将在本文中坚持这一点。

在 Ubuntu Linux 上安装 TensorFlow

在开始设置 TensorFlow 之前,您需要在 Ubuntu 上启用 Universe 存储库。 您可以使用以下命令执行此操作:

sudo add-apt-repository universe

或前往 软件和更新 选项并从那里启用它:

第一步:获取Python开发环境

首先,您必须设置 Python 开发环境才能继续。 默认情况下,您应该在 Ubuntu 上安装 Python 3.6.x。 您可以使用以下命令检查 Python 版本:

python3 --version

确保你有 蟒蛇 3.5–3.7 (按照官方指示)。 如果您没有它或需要升级它,请在终端中使用以下命令安装它:

sudo apt install python3-dev

第 2 步:获取点子

接下来,您需要检查您是否有 Python 的 pip 包管理器.

pip3 --version

如果您没有安装它,您可以通过键入以下内容来安装它:

sudo apt install python3-pip

但是,您可能没有板载最新版本。

笔记: TensorFlow 需要 pip 19.0 或以上版本. 不用担心,一旦在接下来的步骤中设置了虚拟环境,您就可以升级 pip。

第 3 步:设置 Python 虚拟环境

现在,您已经完成了 Python 和 pip 的设置,您需要设置一个 Python开发的虚拟环境. 这样,无论您使用 Python 设置做什么,都不会影响主机系统。

首先,安装 virtualenv:

sudo -H pip3 install -U virtualenv

-H 标志 将 HOME 环境变量设置为主目录。

你会想要 创建 Python 虚拟环境 通过指定目录并选择 Python 解释器。 这是您必须输入的内容:

virtualenv --system-site-packages -p python3 ./venv

完成后,您只需要激活虚拟环境,为此,请按以下方式使用 source 命令:

source ./venv/bin/activate

当您的虚拟环境处于活动状态时,您会观察到您的 shell 提示符将以 (文) 如下图所示。

Python 虚拟环境

接下来,您需要在不影响主机设置的情况下升级虚拟环境中的 pip 版本(如果它已经是最新的,您将收到通知):

pip install --upgrade pip

如果您好奇,您还可以通过键入以下内容查看安装在虚拟环境中的软件包列表:

pip list

第 4 步:安装 TensorFlow

最后,您将要安装 TensorFlow。

确保您使用的是虚拟环境。 输入以下命令来安装 TensorFlow:

pip install --upgrade tensorflow

您还可以选择通过在虚拟环境中键入以下命令来验证安装:

python -c "import tensorflow as tf;print(tf.reduce_sum(tf.random.normal([1000, 1000])))"

最后,您已经在 Ubuntu 上成功安装了 TensorFlow!

在 Pop!_OS 上安装 TensorFlow

与 Ubuntu 不同,如果您有 Pop!_OS,则无需执行所有这些步骤,只需一个命令即可使用基本系统 python。

您只需要在终端中输入以下命令:

sudo apt install tensorflow-cuda-latest

有关如何使用它的更多信息,您可以参考 官方 Pop!_OS 资源.

总结

请记住,无论何时要在 Ubuntu 上使用 TensorFlow,都必须先进入指定的 Python 虚拟环境。

如果您是 TensorFlow 新手,您可能需要查看 官方资源 可供学习。

我希望这能够帮到你。 还值得注意的是,我执行了在全新 Ubuntu 安装上安装 TensorFlow 的步骤——因此,如果您在安装中修改了某些内容,则在 Ubuntu 上安装 TensorFlow 时可能需要进行一些故障排除。

请随时在下面的评论中告诉我您的想法。