码迷,mamicode.com
首页 > 系统相关 > 详细

在 Ubuntu 上安装 TensorFlow (官方文档的翻译)

时间:2018-01-22 22:58:53      阅读:323      评论:0      收藏:0      [点我收藏+]

标签:pre   guide   body   详细信息   应用   需要   版本   分析   mon   

 

本指南介绍了如何在 Ubuntu 上安装 TensorFlow。这些指令也可能对其他 Linux 变体起作用, 但是我们只在Ubuntu 14.04 或更高版本上测试了(我们只支持)  这些指令。

 

一、确定安装 TensorFlow

1、确定要安装的 TensorFlow
要安装以下类型的 TensorFlow, 必须选择其中一种:
  • 仅使用 CPU 支持的 TensorFlow。如果您的系统没有 NVIDIA? GPU, 则必须安装此版本。请注意, 此版本的 TensorFlow 通常更容易安装 (通常在5或10分钟), 因此, 即使您有 NVIDIA? GPU, 我们建议先安装此版本。
  • GPU 支持的TensorFlow。TensorFlow 程序在 GPU 上运行的速度通常比在 CPU 上快得多。因此, 如果您的系统有一个 NVIDIA? GPU 满足下面所示的先决条件, 并且您需要高效运行应用程序, 则应最终安装此版本。

2、使用 GPU 支持的 TensorFlow  要求

如果使用指南介绍一种机制安装 GPU 支持的 TensorFlow , 则必须系统安装以下 NVIDIA 软件:

  • CUDA? Toolkit 8.0。有关详细信息, 请参阅 NVIDIA 的文档。请确保将相关的 Cuda 路径名追加到 LD_LIBRARY_PATH 环境变量中, 如 NVIDIA 文档中所述。
  • 与CUDA? Toolkit 8.0 关联的 NVIDIA 驱动程序。
  • cuDNN v6.0。有关详细信息, 请参阅 NVIDIA 的文档。确保按照 NVIDIA 文档中的描述,创建 CUDA_HOME 环境变量。
  • 具有 CUDA 计算能力3.0 或更高的 GPU 卡。有关支持的 GPU 卡的列表, 请参见 NVIDIA 文档
  •  libcupti-dev库, 这是 NVIDIA CUDA 配置文件的工具界面。此库提供高级性能分析支持。安装, 请发出以下命令:
sudo apt-get install libcupti-dev

 

如果您有更早期版本的软件包, 请升级到指定的版本。如果无法升级,可以使用 GPU 支持的 TensorFlow, 必须执行以下操作:
  • 从源码中安装 TensorFlow, 如从源码安装 TensorFlow
  • 至少安装升级到以下 NVIDIA 版本:
    •   CUDA toolkit 7.0或更高
    •   cuDNN v3 更大
    •   具有 CUDA 计算能力3.0 或更高的 GPU 卡。

 

 3、确定如何安装 TensorFlow

您必须选择安装 TensorFlow 的方式。支持的方式如下所示:
  • virtualenv
  • "本地" pip

1)用 virtualenv安装

执行以下步骤 Virtualenv 安装 TensorFlow:

 

(1)通过发出以下命令之一安装 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

(2)通过发出以下命令之一创建 virtualenv 环境:

 virtualenv --system-site-packages targetDirectory # for Python 2.7
 virtualenv --system-site-packages -p python3 targetDirectory # for Python 3.n

其中 targetDirectory 指定 virtualenv tree的顶部。我们的指示假定 targetDirectory 是~/ tensorflow, 但您可以选择任何目录。

(3)通过发出以下命令之一激活 virtualenv 环境:

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

源代码命令前面提示将更改以下内容:

(tensorflow)$ 

(4)确保已安装 pip ≥8.1:

(tensorflow)$ easy_install -U pip

(5)在激活的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。

(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 包的 URL。tfBinaryURL  取决于操作系统、Python 版本和 GPU 。在这里为您的系统找到合适的 tfBinaryURL 。例如, 如果安装 LinuxPython 3.4 支持 CPU 的 TensorFlow发出以下命令在激活 virtualenv 环境安装 TensorFlow:

(tensorflow)$ pip3 install --upgrade  https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-1.4.1-cp34-cp34m-linux_x86_64.whl

如果遇到安装问题, 请参阅常见安装问题

 

2)后续步骤
安装 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 用, 可以通过调用停用函数环境, 下所示:

(tensorflow)$ deactivate 

提示符将恢复到您的默认提示 (由 PS1 环境变量定义)。

 

3)卸载 TensorFlow
要卸载 TensorFlow, 只需删除您创建的树。例如:
$ rm -r targetDirectory 

  

 

 

-------------------------------------

在 Ubuntu 上安装 TensorFlow (官方文档的翻译)

标签:pre   guide   body   详细信息   应用   需要   版本   分析   mon   

原文地址:https://www.cnblogs.com/hezhiyao/p/8331333.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!