Centos7上源码编译tensorflow

tensorflow官网:

https://www.tensorflow.org/install/source?hl=zh_cn

编译前概述:

本人编译的是tensorflowr1.13版本,据我所知,tensorflow每一个版本对构建工具都有所要求,需得与对应版本进行匹配。1.13版本所需要的工具版本为:

gcc5.3.0、python3.5及以上、cuda10.0、bazel0.21.0

最后就是,不管是编译哪个版本,都要有充分的信心,编译的过程中遇到一些奇奇怪怪的错误、在所难免。

安装工具:

1.1、安装python3

    参考文档:https://blog.csdn.net/lovefengruoqing/article/details/79284573

1.2、python3-dev、python3-pip

    sudo apt install python3-dev python3-pip

1.3、安装gcc5.3.0

       1.3.1、 sudo yum install libmpc-devel mpfr-devel gmp-devel

       1.3.2、curl ftp://ftp.mirrorservice.org/sites/sourceware.org/pub/gcc/releases/gcc-5.3.0/gcc-5.3.0.tar.bz2 -O

       1.3.3、tar xvfj gcc-5.3.0.tar.bz2

       1.3.4、cd gcc-5.3.0
                  ./configure --prefix=/usr/local/gcc-5.3.0 --enable-languages=c,c++ --disable-multilib
                  make -j 16     //j选项数字表示,你的cpu核数x2,目的是提高编译速度
                  make install

       1.3.5、export LD_LIBRARY_PATH=/usr/local/gcc-5.3.0/lib64:/usr/local/lib:$LD_LIBRARY_PATH
                  export PATH=/usr/local/gcc-5.3.0/bin:/usr/local/bin:$PATH

1.4、pip 软件包依赖项

    pip install -U --user pip six numpy wheel setuptools mock future>=0.17.1
    pip install -U --user keras_applications==1.0.6 --no-deps    
    pip install -U --user keras_preprocessing==1.0.5 --no-deps    

1.5、安装bazel(0.21.0)

    sudo yum remove bazel    ---卸载原先的bazel版本
    # 从 https://github.com/bazelbuild/bazel/releases 下载对应release安装   
    wget https://github.com/bazelbuild/bazel/releases/download/0.21.0/bazel-0.21.0-installer-linux-x86_64.sh 
    chmod +x bazel-0.21.0-installer-linux-x86_64.sh
    ./bazel-0.21.0-installer-linux-x86_64.sh --user

    注:--user 参数安装Bazel在系统的 $HOME/bin 目录并设置 .bazelrc 路径到 $HOME/.bazelrc ,

           所以需要在 ~/.bashrc 中添        加 export PATH="$PATH:$HOME/bin"

1.6、CUDA10.0安装

    可以去到官网里下载安装,这里因为运维事先已经安装好了。

    nvidia-smi命令可以查看gpu使用情况

下载tensorflow源代码:

    git clone https://github.com/tensorflow/tensorflow.git
    cd tensorflow
    git branch -a 查看远程分支
    git checkout r1.13 切换到指定分支

编译tensorflow:

    2.1 配置编译环境

          执行命令: ./configure

         

编译pip软件包(gpu):

bazel build --config=opt --cxxopt=-fabi-version=6 --incompatible_package_name_is_a_function=false --config=cuda --verbose_failures  //tensorflow/tools/pip_package:build_pip_package

编译c++软件版本(gpu):

bazel build --config=opt --cxxopt=-fabi-version=6 --incompatible_package_name_is_a_function=false --config=cuda --verbose_failures //tensorflow:libtensorflow_cc.so

编译成功截图:

安装头文件(可选项,用于tensorflow c++开发):

 

mkdir /usr/local/include/tf
cp -r bazel-genfiles/ /usr/local/include/tf/
cp -r tensorflow /usr/local/include/tf/
cp -r third_party /usr/local/include/tf/
cp -r bazel-bin/tensorflow/libtensorflow_cc.so /usr/local/lib/

 

 

 

 

          

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值