跳转至

编译tensorflow

为什么我们要去编译tensorflow呢? pip安装不香么? 那是因为有些cpu的指令集可能和pip安装的

tensorflow的不一致, 这个时候可能会出现core dump的问题, 因此我们需要从新编译tensorflow.

安装必要的依赖

假设我们已经安装了python:

配置pip源:

mkdir~/.pip 
echo"[global]">>   ~/.pip/pip.conf
echo"index-url=https://pypi.tuna.tsinghua.edu.cn/simple">> ~/.pip/pip.conf
pipinstall   pipsixnumpywheelsetuptoolsmockfuture>=0.17.1
pipinstall keras_applications==1.0.6--no-deps 
pipinstall keras_preprocessing==1.0.5--no-deps

安装java

apt-getinstallopenjdk-8-jdk

如果安装oracle的jdk, 也可以下载oracle jdk对应的版本.

安装bazel

./root/data/bazel-0.20.0-installer-linux-x86_64.sh--user 
echo'exportPATH=$PATH:/root/bin'>>/etc/profile 
source/etc/profile

编译tensorflow

我们需要去git上把tensorflow的源代码给clone下来:

cd/thd 
gitclonehttps://github.com/tensorflow/tensorflow.git 
cdtensorflow 
gitcheckout--trackorigin/r1.12 
./configure

这时候会有一堆需要选项, 根据需要去选就可以了.

不过这时候可能会有cuda版本安装的问题, 因为tensorflow对于不同版本的cuda是有要求的, 因此我们需

要安装多个版本的cuda.

Document