如果你喜欢机器学习,你可能需要利用 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 提示符将以 (文) 如下图所示。
接下来,您需要在不影响主机设置的情况下升级虚拟环境中的 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 时可能需要进行一些故障排除。
请随时在下面的评论中告诉我您的想法。