如何在Ubuntu 18.04上安装TensorFlow GPU(分步指南)

Ubuntu安装TensorFlow GPU教程介绍TensorFlow是一个开源机器学习平台。谷歌设计该软件是为了帮助研究人员、数据科学家和开发人员简化实现机器学习模型的过程。
Ubuntu如何安装TensorFlow GPU?这个用于数值计算的端到端库可以在多个 CPU、GPU 以及移动操作系统上运行。在本教程中,了解如何在 Ubuntu 18.04 上安装到 TensorFlow。

如何在Ubuntu 18.04上安装TensorFlow GPU(分步指南)

文章图片
如何在Ubuntu安装TensorFlow GPU
注意:对于 CentOS,请参阅我们在 CentOS上安装 TensorFlow的指南。
先决条件
  • Ubuntu Linux 系统(16.04 或更高版本)
  • 访问命令行/终端窗口 (  Ctrl  +  Alt  +  T  )
  • 具有sudo  权限的用户帐户 
  • 点 19.0 或更高版本
第 1 步:安装所需的软件包如何在Ubuntu安装TensorFlow GPU?在安装 TensorFlow 之前,你需要设置 Python 开发环境。它包括以下软件:
  • Python(3.4 或更高版本)
  • 在PIP包管理器(不超过19.0版本以上)
  • Virtualenv(用于隔离 Python 环境的软件)
打开终端窗口并开始更新存储库:
sudo apt update

接下来,使用以下命令安装前两个软件包:
sudo apt install python3-dev python3-pip

如何在Ubuntu 18.04上安装TensorFlow GPU(分步指南)

文章图片
【如何在Ubuntu 18.04上安装TensorFlow GPU(分步指南)】然后,为 Virtualenv 的广泛系统安装运行以下命令:
sudo pip3 install -U virtualenv

如何在Ubuntu 18.04上安装TensorFlow GPU(分步指南)

文章图片
Ubuntu安装TensorFlow GPU教程
步骤 2:创建虚拟环境Ubuntu如何安装TensorFlow GPU?现在你的 Ubuntu 系统上有 Virtualenv,你可以使用它来创建和隔离 Python 环境。
在新的./venv目录中创建你的第一个环境:
virtualenv --system-site-packages -p python3 ./venv

如何在Ubuntu 18.04上安装TensorFlow GPU(分步指南)

文章图片

然后,激活虚拟环境以开始在其中工作。运行以下命令:
source ./venv/bin/activate

你的 shell 提示现在应该有一个(venv)前缀,如下图所示:
如何在Ubuntu 18.04上安装TensorFlow GPU(分步指南)

文章图片
激活venv 后,继续在隔离环境中安装pip:
pip install --upgrade pip

如何在Ubuntu 18.04上安装TensorFlow GPU(分步指南)

文章图片
如果要查看虚拟环境中所有包的完整列表,请使用以下命令:
pip list

它显示所有软件包及其各自的版本,如下图所示:
如何在Ubuntu 18.04上安装TensorFlow GPU(分步指南)

文章图片
Ubuntu安装TensorFlow GPU教程
第 3 步:安装 TensorFlowUbuntu如何安装TensorFlow GPU?下一步取决于你是安装TensorFlow for CPU还是TensorFlow for GPU。选择取决于你的工作负载和硬件选项的性质。
选项 1:为 CPU 安装 TensorFlow
默认的TensorFlow软件包支持基于 CPU 的工作负载。要安装包及其依赖项,请键入以下命令:
pip install --upgrade tensorflow

选项 2:为 GPU 安装 TensorFlow
Ubuntu安装TensorFlow GPU教程:TensorFlow for GPU需要专用的 NVIDIA CUDA?-enabled GPU 和相关驱动程序。该软件包适用于基于 GPU 的机器学习工作负载。
要为 GPU安装最新的稳定版 TensorFlow  ,请运行以下命令:
pip install --upgrade tensorflow-gpu

有关 TensorFlow for GPU 的硬件和软件要求列表,请参阅TensorFlow 关于 GPU 支持的文档。
选项 3:安装旧版本的 TensorFlow
用于 CPU 和 GPU 的旧版TensorFlow也可供下载。
如何在Ubuntu安装TensorFlow GPU?通过以以下格式运行命令来安装1.14 及更早版本:
pip install tensorflow==package_version

要为 CPU 1.14安装TensorFlow,请运行以下命令:
pip install tensorflow==1.14

要为 GPU 1.14安装TensorFlow,请运行以下命令:
pip install tensorflow-gpu==1.14

1.15 版支持基于 CPU 和 GPU 的工作负载。要安装TensorFlow 1.15,请输入以下命令:
pip install tensorflow-gpu==1.15rc2

第 4 步:验证 TensorFlow 安装要在 Ubuntu 中验证 TensorFlow 的安装,请在终端窗口中输入命令:
python -c "import tensorflow as tf; print(tf.reduce_sum(tf.random.normal([ 1000, 1000])))"

Ubuntu安装TensorFlow GPU教程结论Ubuntu如何安装TensorFlow GPU?本文向你展示了在 Ubuntu 18.04 上安装 TensorFlow 的步骤。系统上安装了所需的TensorFlow 版本(CPU 或 GPU 支持)后,你现在可以继续开发机器学习模型。

    推荐阅读