2016-06-28 49 views
9

Ho un progetto C++ impostato in CLion che utilizza CMake. Sto usando varie librerie di terze parti e vorrei anche integrare Tensorflow.utilizzare tensorflow precompilato con cmake

Ho provato Bazel per compilare tensorflow ad una libreria condivisa libtensorflow.so che tipo di lavorato tuttavia ci sono ancora un bel paio di dipendenze (ad esempio, a una versione protobuf corrente e una volta che faccio che ci sono più), che Dovrei sistemare.

C'è un modo per utilizzare il repository Git standard Tensorflow e in qualche modo collegare le librerie precompilate per l'utilizzo Python? O c'è un altro modo conveniente?

Tensorflow in Python funziona bene per me.

risposta

4

Sono consapevole che questa risposta è abbastanza tardi, ma ho riscontrato il tuo problema esatto ed è stato in grado di risolverlo. Ho creato un repository here che descrive come realizzare esattamente ciò che desideri. L'essenza è:

  • Aggiungi una regola di costruzione al repository TensorFlow contenente tutti gli elementi C++ richiesti.
  • Costruisci la libreria condivisa usando Bazel e copia tutte le intestazioni su /usr/local.
  • Installare versioni specifiche di Protobuf ed Eigen (questo viene fatto automaticamente con gli script) o aggiungerle come dipendenze esterne.
  • Configura il tuo progetto CMake con i file specificati.

Se avete domande o problemi, non esitate a contattarmi.

+0

Perché è necessario inserire i file in/usr/local? Posso semplicemente mettere i miei file da qualche altra parte, e fare in modo che Cmake li trovi direttamente? Grazie per l'aiuto con cmake – Pototo