Ubuntu 16.04 build TensorFlow & Lite C++ 动态连接库

首先要安装bazel tool

下载源码:git clone https://github.com/tensorflow/tensorflow.git

Build Tensorflow:

bazel build  --jobs 6 //tensorflow:libtensorflow_cc.so 

假如需要增加Intel CPU 加速:

bazel build --config=mkl --config=opt  --jobs 6 //tensorflow:libtensorflow_cc.so 

Build Tensorflow Lite:

编辑TensorFlow source code里面的"tensorflow/contrib/lite/BUILD"文件,增加:

cc_binary(
    name = "libtflite.so",
    deps = [":framework",
        "//tensorflow/contrib/lite/kernels:builtin_ops",
        "//tensorflow/contrib/lite/kernels:eigen_support",
        "//tensorflow/contrib/lite/kernels:gemm_support",
    ],
    linkopts=["-shared -Wl,--whole-archive" ],
    linkshared=1

)

bazel build  --jobs 6 //tensorflow/contrib/lite:libtflite.so  --cxxopt="-std=c++11" --copt="-O3"

使用这些C++库的link的时候最好加上:LDFLAGS +=  -Wl,--allow-multiple-definition -Wl,--whole-archive -ltflite

Build TensorFlow pip install文件,比较慢,一般不推荐,自己build主要是增加本地的优化配置选项,大家可以自行增加:

bazel build --jobs 6 //tensorflow/tools/pip_package:build_pip_package

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

pip install --upgrade /tmp/tensorflow_pkg/tensorflow-*.whl

阅读更多
个人分类: tools
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

不良信息举报

Ubuntu 16.04 build TensorFlow & Lite C++ 动态连接库

最多只允许输入30个字

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭