centos7 源码编译安装TensorFlow CPU 版本

一、前言

我们都知道,普通使用pip安装的TensorFlow是万金油版本,当你运行的时候,会提示你不是当前电脑中最优的版本,特别是CPU版本,没有使用指令集优化会让TensorFlow用起来更慢。

但是在编译之中,发现很多坑,由此记录一下。

环境相关:

系统:centos7

python版本:2.7.5

二、准备

1. 安装相关依赖

# 一般会缺失的依赖
yum -y install java-1.8.0-openjdk-devel automake autoconf libtool libicu gcc-c++

2. pip安装相关库

# python 相关,连接境外源较慢,建议更换成国内源,比如 douban 等
pip install --upgrade pip
pip install numpy grpcio Keras-Applications Keras-Preprocessing h5py requests enum

3. 查看可优化的信息

直接使用pip安装TensorFlow并运行随便一个TensorFlow相关程序,如果出现以下内容则需要优化相关指令集:

将这些信息记录下来,后面会用到。

4. 安装Bazel

Bazel是编译TensorFlow的工具,首先下载repo文件:

进入网页 https://copr.fedorainfracloud.org/coprs/vbatts/bazel/ 选择自己系统的版本并下载,我的系统是centos7,直接执行命令:

wget https://copr.fedorainfracloud.org/coprs/vbatts/bazel/repo/epel-7/vbatts-bazel-epel-7.repo

复制到yum.repos.d目录并安装:

cp vbatts-bazel-epel-7.repo /etc/yum.repos.d/
yum install bazel

5. 下载TensorFlow源码

输入命令:

git clone --recurse-submodules https://github.com/tensorflow/tensorflow

如果没安装git,直接安装一下就行了:

yum -y install git

准备工作到此结束。

三、编译

1. 配置编译

进入源码目录并配置:

cd tensorflow/
./configure

刚开始会出现让你选择python的一些路径的,输入正确的安装目录就行了。

然后会出现一堆“Do you wish to build TensorFlow with…”的提示,安装自己的情况去选择,不确定的直接回车就行了,不过因为编译的是CPU版本,所以GPU那里需要选n,其他根据实际情况去选择。

最后是一堆配置选择,也是直接回车,默认就可以了。

2. 开始编译

根据pip安装的TensorFlow,运行时提示的优化信息,作为优化参数,根据自己的情况输入命令:

bazel build -c opt --copt=-msse4.1 --copt=-msse4.2 //tensorflow/tools/pip_package:build_pip_package

注意这里的优化参数一定要填正确,要不然CPU不支持,但编译了,到时候用的时候有可能会报错。注意这是一个很慢很慢的过程,请耐心等待。编译界面如下。

不过编译期间当然会出现各种错误,譬如我就遇到找不到python相关的头文件的,最后发现都在/usr/include/python2.7/ 目录里,而代码中指向的是/usr/include/,所以只得把python2.7目录里的头文件复制一份到母目录。此外还有很多错误,这个只能根据自己的系统和环境,一个个去解决了,此处省略一万字。

ps:如果编译一般发现之前的参数错了,先clean一下编译输出文件再编译:

bazel clean

不然会出现很奇怪的问题。

如果顺利,编译会经历一个极其漫长的过程,可以先泡杯茶喝喝。

3. 创建whl文件

编译完成后在tensorflow下会创建很多符号连接,譬如bazel-bin,运行以下命令创建whl文件:

bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg

输出如下图所示:

至此,编译并生成whl过程才算完成,最后得到文件tensorflow-1.12.0rc0-cp27-cp27mu-linux_x86_64.whl,根据不同的平台名字会不大一样。

四、安装

1. 卸载直接用pip直接安装的版本

pip uninstall tensorflow

2. 使用pip安装之前生成的whl文件

pip install /tmp/tensorflow_pkg/tensorflow-1.12.0rc0-cp27-cp27mu-linux_x86_64.whl

如无意外,就安装完成了,输出如下图所示:

3. 测试

运行TensorFlow相关代码,如果没有优化提示,证明一切正常。

>>> import tensorflow as tf
>>> hello = tf.constant(‘Hello, TensorFlow!‘)
>>> sess = tf.Session()
>>> print sess.run(hello)
Hello, TensorFlow!
>>> a = tf.constant(10)
>>> b = tf.constant(32)
>>> print sess.run(a+b)
42
>>> exit()

五、参考

1. CentOS源码编译安装Tensor Flow

2. Tensorflow Serving CentOS 7源码编译(CPU 版)

(完)

原文地址:https://www.cnblogs.com/harrymore/p/10028489.html

时间: 2024-10-13 07:02:54

centos7 源码编译安装TensorFlow CPU 版本的相关文章

源码编译安装Mysql5.7版本

1.源码编译安装mysql(5.7版本) 下载地址: https://cdn.mysql.com//Downloads/MySQL-5.7/mysql-boost-5.7.18.tar.gz #(MySQL从5.7版本之后,boost是必须的,建议把系统自带的boost库卸载,源码编译安装高版本 Boost库是为C++语言标准库提供扩展的一些C++程序库的总称) (1)环境准备: [[email protected] ~]# ls mysql-boost-5.7.18.tar.gz  nginx

CentOS7源码编译安装FreeRadius3.17

(一)环境简介1.OS:Centos7.42.软件:Freeradiusd3.17(ftp://ftp.freeradius.org/pub/freeradius/freeradius-server-3.0.17.tar.gz)3.网站:https://freeradius.org/ (二)安装1.yum安装所需的依赖包[[email protected] raddb]#yum install openssl openssl-devel libtalloc-devel libtalloc* -y

CentOS7 源码编译安装Tengine

简介 Tengine是由淘宝网发起的Web服务器项目.它在Nginx的基础上,针对大访问量网站的需求,添加了很多高级功能和特性.它的目的是打造一个高效.安全的Web平台. 发展 Tengine的性能和稳定性已经在大型的网站如淘宝网,天猫商城等得到了很好的检验. 从2011年12月开始,Tengine成为一个开源项目,Tengine团队在积极地开发和维护着它.Tengine团队的核心成员来自于淘宝.搜狗等互联网企业.Tengine是社区合作的成果,我们欢迎大家参与其中,贡献自己的力量. 一.安装编

CentOS7源码编译安装PHP7.2

一.环境 CentOS7 二.相关资源 PHP官方网站 PHP官方下载页 三.编译安装 1. 下载php 下载并解压 # 下载php wget https://www.php.net/distributions/php-7.2.16.tar.gz # 解压 tar -zxvf php-7.2.16.tar.gz 查看目录 [[email protected] ~]# ll 总用量 19232 drwxrwxr-x 14 root root 4096 3月 5 19:05 php-7.2.16 -

centos7 源码编译安装 nginx

安装步骤 下载 nginx 源码包 官网 $ wget http://nginx.org/download/nginx-1.16.0.tar.gz 解压 nginx 压缩包 $ tar -zxvf nginx-1.16.0.tar.gz 运行 configure 文件检测程序 $ cd nginx-1.16.0 $ ./configure --prefix=/usr/local/nginx checking for OS + Linux 3.10.0-957.12.2.el7.x86_64 x8

centos7 源码编译安装heartbeat 以及结合nginx测试高可用

1.环境CentOS Linux release 7.4.1708 (Core) 3.10.0-693.el7.x86_64主(heartbeat27=uname -n)10.0.0.27(ip) 10.0.10.27(心跳) 10.0.0.29(vip)备(heartbeat28=uname -n)10.0.0.28(ip) 10.0.10.28(心跳) 10.0.0.29(vip)2.cat /etc/hosts10.0.0.27 heartbeat2710.0.0.28 heartbeat

centos7 源码编译安装 php

准备工作 下载 PHP 源码包并解压 $ wget https://www.php.net/distributions/php-7.2.19.tar.bz2 $ tar -jxvf php-7.2.19.tar.bz2 进入 PHP 源码包目录 $ cd php-7.2.19 配置和构建 PHP 常用配置项及其说明 如果看着麻烦可以直接看下面的总结步骤 --prefix=/usr/local/php7 # 配置安装目录 --with-config-file-path=/usr/local/php

centos7 源码编译安装nginx教程 nginx安装脚本

安装nginx需要pcre zlib openssl的库,下文都是在官网直接下载用作编译安装 该nginx安装教程,有安装maxmind IP 库 该nginx安装教程有安装openrestry 系统使用了centos 7 该教材有修改最大打开文件描述符数到最大 该教程是nginx安装的shell脚本 #!/bin/bash yum install epel-release -y yum install gcc gcc-c++ make automake autoconf libtool ipt

centos7源码编译安装Ansible详细部署

一.基础介绍==========================================================================================ansible是新出现的自动化运维工具,基于Python开发,集合了众多运维工具(puppet.cfengine.chef.func.fabric)的优点,实现了批量系统配置.批量程序部署.批量运行命令等功能.ansible是基于模块工作的,本身没有批量部署的能力.真正具有批量部署的是ansible所运行的