2016-08-15 40 views

回答

0

據我所知,沒有官方的方式在TF回購之外構建你的代碼。但是,您可以使用tensorflow_cc項目,該項目爲您構建和安裝TF C++庫並提供方便的CMake目標以及所有必需的標題和鏈接參數。

0

是有,你只需要做到以下幾點:

  1. 克隆tensorflow庫

  2. ./configure你的鏈接(啓用CUDA等)

  3. 使用bazel build build //tensorflow:libtensorflow.so//tensorflow:libtensorflow_cc.so

  4. 創建自己的項目,.so庫和鏈接:■你剛編譯,你需要在編譯/鏈接時下列選項:

    -I/path/to/repo/tensorflow 
    -L/path/to/repo/tensorflow/bazel-bin/tensorflow 
    -ltensorflow_framework -ltensorflow_cc 
    
  5. .hpp.cpp文件現在可以使用tensorflow::Session,以及其他類的,包括相應的頭文件:

    #include "tensorflow/core/public/session.h" 
    #include "tensorflow/cc/ops/standard_ops.h" 
    

可能還需要(供protobuf的編譯器生成的protoc報頭):

-I/path/to/repo/tensorflow/bazel-genfiles/ 

以及對依賴庫(例如它抱怨nsync):

-I/path/to/repo/tensorflow/bazel-tensorflow/external/nsync/public 

希望這是有幫助的。