源码编译安装tensorflow

很久没有玩tensorflow,看到新闻1.12版本发布了,心血来潮找个服务器安装一下python

首先服务器

pip3 install -U tensorflow工具

安装顺利,而后启用python3ui

import tensorflow as tf google

提示不支持AVE,这是什么鬼?????ip

在服务器上执行命令,查看一下cpupip

cat /proc/cpuinfo 编译

居然不支持AVE指令集,好吧,公司比较单薄,用的都是古董,看一下tensorflow,1.6版本官方发布的whl包支持AVE了,若是cpu不支持AVE只能本身编译,看来须要打怪了。import

准备工做tensorflow

下载bazel,google的构建编译工具,虽然没用过,可是不复杂。

下载最新的tensorflow1.12

tar -zxvf tensorflow-1.12.0.tar.gz

cd tensorflow-1.12.0

./configure

 查看cpu支持sse4_1和sse4_2

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

而后就是漫长的等待,若是没有错误使用下面的命令生成pip包

 bazel-bin/tensorflow/tools/pip_package/build_pip_package /opt

pip3 install -U /opt/tensorflow.whl

相关文章
相关标签/搜索