Quantcast
Channel: 英特尔开发人员专区文章
Viewing all articles
Browse latest Browse all 583

在英特尔® 架构上构建和安装 TensorFlow*

$
0
0

简介

TensorFlow* 是一款领先的深度学习和机器学习框架。截至 2017 年 5 月,该框架集成了面向英特尔® 至强® 处理器和英特尔® 至强融核™ 处理器的多项优化。本文是教程系列的第一部分,旨在提供相关信息帮助开发人员通过 GitHub* 存储库中可用的资源构建、安装和探索使用在英特尔® 架构上优化的 TensorFlow*。

资源

The TensorFlow 网站提供有关各组件的信息概述、教程和技术信息,是了解该框架的主要来源。感兴趣的开发人员可从此处着手,全面了解 TensorFlow 在深度学习领域的优势。

文章 TensorFlow 基于现代英特尔® 架构的优化介绍了具体的图形优化、性能试验,以及借助 CPU 优化构建和安装 TensorFlow 的详细信息。强烈建议开发人员阅读这篇文章,详细了解如何面向不同的拓扑结构全面优化 TensorFlow,以及进行优化后所实现的性能提升。

安装概述

本文所述的安装步骤源于 TensorFlow 网站中的 通过资源安装 TensorFlow指南提供的信息。下文中的步骤仅供大致了解安装流程;由于第三方信息可能随时更改,因此建议您同时回顾 TensorFlow 网站提供的信息。

本文所提供的安装指南仅针对安装带有 CPU 支持的 TensorFlow。目标操作系统和 Python* 分发版分别为 Ubuntu* 16.04 和 Python 2.7。

安装 Bazel* 构建工具

Bazel* 是 Google* 公开提供的构建工具。如果系统已安装 Bazel,请跳过此部分。如果尚未安装,请输入以下指令添加 Bazel 分发版 URI、执行安装并更新系统中的 Bazel:

echo "deb [arch=amd64] http://storage.googleapis.com/bazel-apt stable jdk1.8" | sudo tee /etc/apt/sources.list.d/bazel.list
sudo apt install curl
curl https://bazel.build/bazel-release.pub.gpg | sudo apt-key add -
sudo apt-get update && sudo apt-get install bazel
sudo apt-get upgrade bazel

安装 Python* 相关性

如果系统已安装 Python* 相关性,请跳过此部分。输入以下命令安装 Python 2.7 所需的软件包:

sudo apt-get install python-numpy python-dev python-pip python-wheel

构建安装所需的 TensorFlow* Pip 软件包

如果系统未安装程序 Git*,请发布以下命令:

sudo apt install git

发布以下命令,克隆 GitHub 存储库:

git clone https://github.com/tensorflow/tensorflow

克隆期间创建的 tensorflow目录中包含一个名为 configure的脚本,该脚本必须在创建 pip 软件包和安装 TensorFlow 之前执行。该脚本支持您识别路径名称、相关性和其他构建配置选项。对于在英特尔架构上优化的 TensorFlow,该脚本还支持您设置英特尔® 数学核心函数库(英特尔® MKL)的相关环境设置。执行以下命令:

cd tensorflow
./configure

重要:选择 ‘Y’ 构建支持英特尔 MKL 的 TensorFlow,并选择 ‘Y’ 从 Web 下载 MKL LIB。选择其他配置参数的默认设置。脚本运行完成后,发布以下命令构建 pip 软件包:

bazel build --config=mkl --copt="-DEIGEN_USE_VML" -c opt //tensorflow/tools/pip_package:build_pip_package
bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg

安装 TensorFlow—本地 Pip 选项

此时,新创建的 pip 软件包位于 tmp/tensorflow_pkg之中。接下来安装 TensorFlow,既可采用本地 pip 安装,也可在下文所述的 Anaconda* 虚拟环境中完成。如果采用本地 pip 安装,只需输入以下命令:

sudo pip install /tmp/tensorflow_pkg/tensorflow-1.2.0rc1-cp27-cp27mu-linux_x86_64.whl

(注:本轮的名称(上述斜体)可能因特定构建的不同而有差异。)

完成上述步骤后,务必在进行下一步之前验证安装。注:运行链接中提供的 Python 验证脚本时,务必换成不同的目录,例如:

cd ..

安装 TensorFlow—Conda* 环境选项

注:如果系统已安装 Anaconda ,请跳过此部分。

下载页面中下载 Anaconda,并按照说明运行安装程序脚本。(在本教程中,我们使用的 Anaconda 为 64 位 x86 Python 2.7 版本。)在安装过程中,你需要同意许可协议,选择默认设置,并选择 'yes'以将 Anaconda 添加到路径中。安装完成后,关闭终端,然后打开一个新的终端。

接下来我们将通过 tmp/tensorflow_pkg中新创建的 pip 软件包创建 conda 环境和安装 TensorFlow。运行以下命令,创建 TensorFlow 环境 "inteltf",并发布以下命令:

conda create -n inteltf
source activate inteltf
pip install /tmp/tensorflow_pkg/tensorflow-1.2.0rc1-cp27-cp27mu-linux_x86_64.whl

(注:本轮的名称(上述斜体)可能因特定构建的不同而有差异。)

source deactivate inteltf

进行下一步之前关闭当前终端,然后打开一个新的终端。

运行网站中的下列 Python 代码,重新启动 inteltf环境并验证 TensorFlow 安装:

source activate inteltf
	python>>> import tensorflow as tf>>> hello = tf.constant('Hello, TensorFlow!')>>> sess = tf.Session()>>> print(sess.run(hello))

如果安装成功,Python 程序将输出 “Hello, TensorFlow!”。

后续文章

本教程系列的下一篇文章将介绍如何安装 TensorFlow Serving* — 专为生产环境设计的高性能机器学习服务系统。


Viewing all articles
Browse latest Browse all 583

Trending Articles



<script src="https://jsc.adskeeper.com/r/s/rssing.com.1596347.js" async> </script>