Ubuntu安装TensorFlow GPU教程介绍
TensorFlow是一个开源机器学习平台。谷歌设计该软件是为了帮助研究人员、数据科学家和开发人员简化实现机器学习模型的过程。
Ubuntu如何安装TensorFlow GPU？这个用于数值计算的端到端库可以在多个 CPU、GPU 以及移动操作系统上运行。在本教程中，了解如何在 Ubuntu 18.04 上安装到 TensorFlow。
注意：对于 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
然后，为 Virtualenv 的广泛系统安装运行以下命令：
sudo pip3 install -U virtualenv
步骤 2：创建虚拟环境
Ubuntu如何安装TensorFlow GPU？现在你的 Ubuntu 系统上有 Virtualenv，你可以使用它来创建和隔离 Python 环境。
在新的./venv目录中创建你的第一个环境：
virtualenv --system-site-packages -p python3 ./venv
然后，激活虚拟环境以开始在其中工作。运行以下命令：
source ./venv/bin/activate你的 shell 提示现在应该有一个(venv)前缀，如下图所示：
激活venv 后，继续在隔离环境中安装pip：
pip install --upgrade pip
如果要查看虚拟环境中所有包的完整列表，请使用以下命令：
pip list它显示所有软件包及其各自的版本，如下图所示：
第 3 步：安装 TensorFlow
Ubuntu如何安装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.141.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 支持）后，你现在可以继续开发机器学习模型。
