在 Ubuntu 上安装 TensorFlow

最新推荐文章于 2024-08-19 16:13:45 发布

hehe_heh

最新推荐文章于 2024-08-19 16:13:45 发布

阅读量698

点赞数

分类专栏：备忘

备忘专栏收录该内容

34 篇文章 1 订阅

订阅专栏

本指南将介绍如何在 Ubuntu 上安装 TensorFlow。虽然这些说明可能也适用于其他 Linux 版本，但我们只在满足以下要求的计算机上验证过这些说明（而且我们只支持在此类计算机上按这些说明操作）：

64 位台式机或笔记本电脑
Ubuntu 16.04 或更高版本

确定要安装哪种 TensorFlow

您必须从以下 TensorFlow 类型中选择其一来进行安装：

仅支持 CPU 的 TensorFlow。如果您的系统没有 NVIDIA® GPU，就必须安装此版本。请注意，此版本的 TensorFlow 通常更容易安装（用时通常在 5 或 10 分钟内），所以即使您拥有 NVIDIA GPU，我们也建议先安装此版本。
支持 GPU 的 TensorFlow。TensorFlow 程序在 GPU 上的运行速度通常要比在 CPU 上快得多。因此，如果您的系统配有满足以下所示先决条件的 NVIDIA® GPU，并且您需要运行性能至关重要的应用，则最终应安装此版本。

运行支持 GPU 的 TensorFlow 所需满足的 NVIDIA 要求

如果您要使用本指南描述的其中一种方式安装支持 GPU 的 TensorFlow，就必须在系统上安装以下 NVIDIA 软件：

CUDA® 工具包 9.0。如需了解详情，请参阅 NVIDIA 文档。请务必按照 NVIDIA 文档中的说明将相关的 CUDA 路径名附加到 LD_LIBRARY_PATH 环境变量中。
与 CUDA 工具包 9.0 相关联的 NVIDIA 驱动程序。
cuDNN v7.0。如需了解详情，请参阅 NVIDIA 文档。请务必按照 NVIDIA 文档中的说明创建 CUDA_HOME 环境变量。
CUDA 计算能力为 3.0 或更高的 GPU 卡（用于从源代码构建），以及 CUDA 计算能力为 3.5 或更高的 GPU 卡（用于安装我们的二进制文件）。如需了解支持的 GPU 卡的列表，请参阅 NVIDIA 文档。
libcupti-dev 库，这是 NVIDIA CUDA 分析工具接口。此库提供高级分析支持。要安装此库，请针对 CUDA 工具包 8.0 或更高版本发出以下命令：
$ sudo apt-get install cuda-command-line-tools
并将其路径添加到您的 LD_LIBRARY_PATH 环境变量中：
$ export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/extras/CUPTI/lib64
对于 CUDA 工具包 7.5 或更低版本，请发出以下命令：
$ sudo apt-get install libcupti-dev

如果您已安装前述软件包的旧版本，请升级到指定版本。如果升级不可行，您仍然可以运行支持 GPU 的 TensorFlow，但前提是您需要执行以下操作：

从源代码安装 TensorFlow（参阅从源代码安装 TensorFlow 中的说明）。
安装或升级到至少以下 NVIDIA 产品版本：
- CUDA 工具包 7.0 或更高版本
- cuDNN v3 或更高版本
- CUDA 计算能力为 3.0 或更高的 GPU 卡

确定如何安装 TensorFlow

您必须选择安装 TensorFlow 的方式。目前可支持如下几种方式：

Virtualenv
“原生”pip
Docker
Anaconda
从源代码安装（详情请参阅这篇单独的指南）。

我们建议采用 Virtualenv 安装方式。 Virtualenv 是一个与其他 Python 开发相互隔离的虚拟 Python 环境，它无法干扰同一计算机上的其他 Python 程序，也不会受其影响。在 Virtualenv 安装过程中，您不仅要安装 TensorFlow，还要安装 TensorFlow 需要的所有软件包。（这一过程其实很简单。）要开始使用 TensorFlow，您只需要“激活”虚拟环境。总而言之，Virtualenv 提供一种安全可靠的机制来安装和运行 TensorFlow。

原生 pip 会直接在您的系统上安装 TensorFlow，而不是通过任何容器系统。我们建议希望多用户系统中的每个用户都能使用 TensorFlow 的系统管理员采用原生 pip 安装。原生 pip 安装并未隔离在单独的容器中进行，因此可能会干扰系统中其他基于 Python 的安装。但是，如果您熟悉 pip 和您的 Python 环境，通常只需一条命令即可进行原生 pip 安装。

Docker 会将 TensorFlow 安装与您计算机上之前就已存在的软件包完全隔离开来。Docker 容器包含了 TensorFlow 及其所有依赖项。请注意，Docker 映像可能比较大（数百 MB）。如果您想要将 TensorFlow 整合到已在使用 Docker 的某个更大的应用架构中，则可以选择 Docker 安装。

在 Anaconda 中，您可以使用 conda 来创建一个虚拟环境。但是，在 Anaconda 内部，我们建议使用 pip install 命令来安装 TensorFlow，而不要使用 conda install 命令。

注意：conda 软件包是由社区提供支持的，并没有任何官方支持。也就是说，TensorFlow 团队既不测试也不维护 conda 软件包。若使用该软件包，您需要自行承担相关风险。

使用 Virtualenv 进行安装

请按照以下步骤使用 Virtualenv 安装 TensorFlow：

发出下列其中一条命令来安装 pip 和 Virtualenv：
$ sudo apt-get install python-pip python-dev python-virtualenv # for Python 2.7 $ sudo apt-get install python3-pip python3-dev python-virtualenv # for Python 3.n
发出下列其中一条命令来创建 Virtualenv 环境：
$ virtualenv --system-site-packages targetDirectory # for Python 2.7 $ virtualenv --system-site-packages -p python3 targetDirectory # for Python 3.n
targetDirectory 用于指定 Virtualenv 树的顶层目录。我们的说明中假定 targetDirectory 为 ~/tensorflow，但您可以选择任何目录。
发出下列其中一条命令来激活 Virtualenv 环境：
$ source ~/tensorflow/bin/activate # bash, sh, ksh, or zsh $ source ~/tensorflow/bin/activate.csh # csh or tcsh
执行上述 source 命令后，您的提示符应该会变成如下内容：
(tensorflow)$
确保安装 pip 8.1 或更高版本：
(tensorflow)$ easy_install -U pip
发出下列其中一条命令以在处于活动状态的 Virtualenv 环境中安装 TensorFlow：
(tensorflow)$ pip install --upgrade tensorflow # for Python 2.7 (tensorflow)$ pip3 install --upgrade tensorflow # for Python 3.n (tensorflow)$ pip install --upgrade tensorflow-gpu # for Python 2.7 and GPU (tensorflow)$ pip3 install --upgrade tensorflow-gpu # for Python 3.n and GPU
如果上述命令执行成功，请跳过第 6 步。如果上述命令执行失败，请执行第 6 步。
（可选）如果第 5 步执行失败（通常是因为您所调用的 pip 版本低于 8.1），请发出以下格式的命令来在活动 Virtualenv 环境中安装 TensorFlow：
(tensorflow)$ pip install --upgrade tfBinaryURL # Python 2.7 (tensorflow)$ pip3 install --upgrade tfBinaryURL # Python 3.n
其中 tfBinaryURL 用于指示 TensorFlow Python 软件包的网址。tfBinaryURL 的具体值取决于操作系统、Python 版本和 GPU 支持。请在这里查找适合您系统的 tfBinaryURL 值。例如，如果您要为装有 Python 3.4 的 Linux 安装仅支持 CPU 的 TensorFlow，可发出以下命令来在活动的 Virtualenv 环境中进行安装：
(tensorflow)$ pip3 install --upgrade \ https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-1.6.0-cp34-cp34m-linux_x86_64.whl

如果您遇到任何安装问题，请参阅常见的安装问题。

后续步骤

安装 TensorFlow 后，请验证安装。

请注意，每次使用 TensorFlow 时，您都必须激活 Virtualenv 环境。如果 Virtualenv 环境当前未处于活动状态，请调用以下命令之一：

$ source ~/tensorflow/bin/activate # bash, sh, ksh, or zsh $ source ~/tensorflow/bin/activate.csh # csh or tcsh

在 Virtualenv 环境激活后，您就可以从这个 shell 运行 TensorFlow 程序。您的提示符将变成如下所示，这表示您的 Tensorflow 环境已处于活动状态：

(tensorflow)$

用完 TensorFlow 后，可以通过发出以下命令调用 deactivate 函数来停用环境：

(tensorflow)$ deactivate

提示符将恢复为您的默认提示符（由 PS1 环境变量所定义）。

卸载 TensorFlow

要卸载 TensorFlow，只需移除您创建的树即可。例如：

$ rm -r targetDirectory

使用原生 pip 进行安装

您可以通过 pip 安装 TensorFlow，安装流程有两种：一种简单，一种复杂一点。

注意：setup.py 的 REQUIRED_PACKAGES 部分中列出了 pip 将要安装或升级的软件包。

先决条件：Python 和 Pip

Ubuntu 中已自动安装 Python。请花点时间确认（通过发出 python -V 命令）您的系统中是否已经安装了下列某个 Python 版本：

Python 2.7
Python 3.4+

Ubuntu 上通常已安装 pip 或 pip3 软件包管理器。请花点时间确认（通过发出 pip -V 或 pip3 -V 命令）是否已经安装 pip 或 pip3。我们强烈建议使用 8.1 或更高版本的 pip 或 pip3。如果未安装 pip 或 pip3 版本 8.1 或更高版本，请发出以下命令来安装 pip 或升级到最新版本：

$ sudo apt-get install python-pip python-dev # for Python 2.7 $ sudo apt-get install python3-pip python3-dev # for Python 3.n

安装 TensorFlow

假设您的 Linux 主机上已安装必备软件，请执行以下步骤：

调用下列其中一条命令来安装 TensorFlow：
$ pip install tensorflow # Python 2.7; CPU support (no GPU support) $ pip3 install tensorflow # Python 3.n; CPU support (no GPU support) $ pip install tensorflow-gpu # Python 2.7; GPU support $ pip3 install tensorflow-gpu # Python 3.n; GPU support
如果前面的命令运行完成，您现在应该验证您的安装。
（可选）如果第 1 步执行失败，请通过发出以下格式的命令来安装最新版本的 TensorFlow：
$ sudo pip install --upgrade tfBinaryURL # Python 2.7 $ sudo pip3 install --upgrade tfBinaryURL # Python 3.n
其中 tfBinaryURL 用于指示 TensorFlow Python 软件包的网址。tfBinaryURL 的具体值取决于操作系统、Python 版本和 GPU 支持。请在这里查找适合您系统的 tfBinaryURL 值。例如，要为装有 Python 3.4 的 Linux 安装仅支持 CPU 的 TensorFlow，可发出以下命令：
$ sudo pip3 install --upgrade \ https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-1.6.0-cp34-cp34m-linux_x86_64.whl
如果此步骤执行失败，请参阅常见安装问题。

后续步骤

安装 TensorFlow 后，请验证您的安装。

卸载 TensorFlow

要卸载 TensorFlow，请发出下列其中一条命令：

$ sudo pip uninstall tensorflow # for Python 2.7 $ sudo pip3 uninstall tensorflow # for Python 3.n

使用 Docker 进行安装

请按照以下步骤通过 Docker 安装 TensorFlow：

按照 Docker 文档中的说明在您的计算机上安装 Docker。
（可选）创建一个名为 docker 的 Linux 组，以支持启动不带 sudo 的容器，具体如 Docker 文档中所述。（如果不执行此操作，则每次调用 Docker 时都必须使用 sudo。）
要安装支持 GPU 的 TensorFlow 版本，您必须先安装 nvidia-docker（存储在 github 中）。
启动包含某个 TensorFlow 二进制映像的 Docker 容器。

本部分的其余内容将介绍如何启动 Docker 容器。

仅支持 CPU

要启动仅支持 CPU（即不支持 GPU）的 Docker 容器，请输入以下格式的命令：

$ docker run -it -p hostPort:containerPort TensorFlowCPUImage

其中：

“-p hostPort:containerPort”是可选项。如果您想从 shell 运行 TensorFlow 程序，请省略此选项。如果您想从 Jupyter Notebook 运行 TensorFlow 程序，请将“hostPort”和“containerPort”设置为 8888。如果您想在容器内部运行 TensorBoard，请再添加一个 -p 标志，并将“hostPort”和“containerPort”设置为 6006。
“TensorFlowCPUImage”是必填项。它用于指示 Docker 容器。请指定如下某个值：
- gcr.io/tensorflow/tensorflow：TensorFlow CPU 二进制映像。
- gcr.io/tensorflow/tensorflow:latest-devel：最新的 TensorFlow CPU 二进制映像以及源代码。
- gcr.io/tensorflow/tensorflow:版本：指定的 TensorFlow CPU 二进制映像版本（如 1.1.0rc1）。
- gcr.io/tensorflow/tensorflow:版本-devel：指定的 TensorFlow GPU 二进制映像版本（如 1.1.0rc1）以及源代码。
gcr.io 是 Google Container Registry。请注意，dockerhub 上也提供了一些 TensorFlow 映像。

例如，以下命令会在 Docker 容器中启动最新的 TensorFlow CPU 二进制映像，您可以通过该容器在 shell 中运行 TensorFlow 程序：

$ docker run -it gcr.io/tensorflow/tensorflow bash

以下命令也可在 Docker 容器中启动最新的 TensorFlow CPU 二进制映像。但是，在这个 Docker 容器里，您可以通过 Jupyter Notebook 运行 TensorFlow 程序：

$ docker run -it -p 8888:8888 gcr.io/tensorflow/tensorflow

Docker 将在您第一次启动 TensorFlow 二进制映像时下载该映像。

GPU 支持

在安装支持 GPU 的 TensorFlow 之前，请确保您的系统满足所有 NVIDIA 软件要求。要启动支持 NVidia GPU 的 Docker 容器，请输入以下格式的命令：

$ nvidia-docker run -it -p hostPort:containerPort TensorFlowGPUImage

其中：

“-p hostPort:containerPort”是可选的。如果您想从 shell 运行 TensorFlow 程序，请省略此选项。如果您想从 Jupyter Notebook 运行 TensorFlow 程序，请将“hostPort”和 containerPort 设置为 8888。
“TensorFlowGPUImage”用于指定 Docker 容器。您必须指定如下某个值：
- gcr.io/tensorflow/tensorflow:latest-gpu：最新的 TensorFlow GPU 二进制映像。
- gcr.io/tensorflow/tensorflow:latest-devel-gpu：最新的 TensorFlow GPU 二进制映像以及源代码。
- gcr.io/tensorflow/tensorflow:版本-gpu：指定的 TensorFlow GPU 二进制映像版本（如 0.12.1）。
- gcr.io/tensorflow/tensorflow:版本-devel-GPU：指定的 TensorFlow GPU 二进制映像版本（如 0.12.1）以及源代码。

我们建议安装最新 (latest) 的某个版本。例如，以下命令会在 Docker 容器中启动最新的 TensorFlow GPU 二进制映像，您可以通过该容器在 shell 中运行 TensorFlow 程序：

$ nvidia-docker run -it gcr.io/tensorflow/tensorflow:latest-gpu bash

以下命令也可在 Docker 容器中启动最新的 TensorFlow GPU 二进制映像。在这个 Docker 容器里，您可以通过 Jupyter Notebook 运行 TensorFlow 程序：

$ nvidia-docker run -it -p 8888:8888 gcr.io/tensorflow/tensorflow:latest-gpu

以下命令会安装旧版 TensorFlow (0.12.1)：

$ nvidia-docker run -it -p 8888:8888 gcr.io/tensorflow/tensorflow:0.12.1-gpu

Docker 将在您第一次启动 TensorFlow 二进制映像时下载该映像。有关详情，请参阅 TensorFlow Docker 自述文件。

后续步骤

您现在应该验证您的安装。

使用 Anaconda 进行安装

请按照以下步骤在 Anaconda 环境中安装 TensorFlow：

按照 Anaconda 下载网站上的说明下载并安装 Anaconda。
调用以下命令创建名为 tensorflow 的 conda 环境，以运行某个版本的 Python：
$ conda create -n tensorflow pip python=2.7 # or python=3.3, etc.
发出以下命令以激活 conda 环境：
$ source activate tensorflow (tensorflow)$ # Your prompt should change
发出以下格式的命令，在您的 conda 环境中安装 TensorFlow：
(tensorflow)$ pip install --ignore-installed --upgrade tfBinaryURL
其中 tfBinaryURL 是 TensorFlow Python 软件包的网址。例如，以下命令将在 Python 3.4 环境中安装仅支持 CPU 的 TensorFlow 版本：
(tensorflow)$ pip install --ignore-installed --upgrade \ https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-1.6.0-cp34-cp34m-linux_x86_64.whl

验证您的安装

要验证您的 TensorFlow 安装，请执行以下操作：

确保您的环境已准备好运行 TensorFlow 程序。
运行一个简短的 TensorFlow 程序。

准备您的环境

如果您是在原生 pip、Virtualenv 或 Anaconda 上进行安装，请执行以下操作：

启动终端。
如果您是通过 Virtualenv 或 Anaconda 进行安装，请激活您的容器。
如果您安装了 TensorFlow 源代码，请导航到未包含 TensorFlow 源代码的任何目录。

如果您是通过 Docker 进行安装，请启动 Docker 容器以运行 bash。例如：

$ docker run -it gcr.io/tensorflow/tensorflow bash

运行一个简短的 TensorFlow 程序

从 shell 中调用 Python，如下所示：

$ python

在 Python 交互式 shell 中输入以下几行简短的程序代码：

# Python
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
如果系统输出以下内容，就说明您可以开始编写 TensorFlow 程序了：

Hello, TensorFlow!

如果您是 TensorFlow 初学者，请参阅 TensorFlow 使用入门。

如果系统输出一条错误消息而不是问候语，请参阅常见的安装问题。

常见的安装问题

我们借助 Stack Overflow 来记录 TensorFlow 安装问题及其补救措施。下表包含了 Stack Overflow 上一些常见安装问题解答的链接。如果您遇到了下表中未列出的错误消息或其他安装问题，请在 Stack Overflow 上进行搜索。如果 Stack Overflow 中没有显示相关错误消息，请在 Stack Overflow 上提一个新的问题，并指定 tensorflow 标签。

Stack Overflow 链接	错误消息
36159194	ImportError: libcudart.so.Version: cannot open shared object file: No such file or directory
41991101	ImportError: libcudnn.Version: cannot open shared object file: No such file or directory
36371137 以及这里	libprotobuf ERROR google/protobuf/src/google/protobuf/io/coded_stream.cc:207] A protocol message was rejected because it was too big (more than 67108864 bytes). To increase the limit (or to disable these warnings), see CodedInputStream::SetTotalBytesLimit() in google/protobuf/io/coded_stream.h.
35252888	Error importing tensorflow. Unless you are using bazel, you should not try to import tensorflow from its source directory; please exit the tensorflow source tree, and relaunch your python interpreter from there.
33623453	IOError: [Errno 2] No such file or directory: '/tmp/pip-o6Tpui-build/setup.py'
42006320	ImportError: Traceback (most recent call last): File ".../tensorflow/core/framework/graph_pb2.py", line 6, in from google.protobuf import descriptor as _descriptor ImportError: cannot import name 'descriptor'
35190574	SSLError: [SSL: CERTIFICATE_VERIFY_FAILED] certificate verify failed
42009190	Installing collected packages: setuptools, protobuf, wheel, numpy, tensorflow Found existing installation: setuptools 1.1.6 Uninstalling setuptools-1.1.6: Exception: ... [Errno 1] Operation not permitted: '/tmp/pip-a1DXRT-uninstall/.../lib/python/_markerlib'
36933958	... Installing collected packages: setuptools, protobuf, wheel, numpy, tensorflow Found existing installation: setuptools 1.1.6 Uninstalling setuptools-1.1.6: Exception: ... [Errno 1] Operation not permitted: '/tmp/pip-a1DXRT-uninstall/System/Library/Frameworks/Python.framework/ Versions/2.7/Extras/lib/python/_markerlib'