锐客网

  1. 首页 > 睿知 > it技术 > >

如何在Ubuntu 18.04上安装TensorFlow GPU（分步指南）

IT知识

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 支持）后，你现在可以继续开发机器学习模型。

    推荐阅读


    上一篇：华为手机滚动截屏怎么操作

    下一篇：22个用于编程和编码的最佳Linux文本编辑器合集