LoginSignup
2
1

More than 3 years have passed since last update.

GPU無しtensorflow 1.14.0 C++におけるUbuntu18.04上でのソースビルドとCMakeで使えるようにする

Last updated at Posted at 2019-09-26

事のきっかっけ

C++でやりたくなった。
(ここに至るまでに、Windowsでのビルドを諦めてしまったのは内緒。)

前提条件

  • OS:Ubuntu 18.04

事前知識

  • .bash_profileに書こうが.bashrcに書こうが特に変わりないので、各自の都合に合わせてください。

ビルド前の事前準備

  • pyenvをインストール。
  1. sudo apt install gitでgitをインストール。
  2. git clone https://github.com/pyenv/pyenv.git ~/.pyenvでpyenvをインストール。

3.vim ~/.bash_profileでエディタを開き、以下の3行を記載し、保存する。

export PYENV_ROOT="$HOME/.pyenv"
export PATH="$PYENV_ROOT/bin:$PATH"
eval "$(pyenv init -)"

4.source ~/.bash_profileを実行し、先程の設定を読み込ませる。

  • pythonを入れる前の下準備

1.sudo apt install zlib1g-dev libssl-dev libbz2-dev libreadline-dev libsqlite3-dev

  • python3.7をインストール
  1. pyenv install 3.7.4を実行し、Python3.7.4をインストール。
  2. pyenv global 3.7.4を実行し、デフォルトを3.7.4にする。
  3. python -Vを実行し、上記のバージョンとなっているのかを確認する。
  • virtualenvをインストール
  1. pip install virtualenvで、インストール。
  • virtualenvで仮想環境の作成
  1. virtualenv py3.7.4で、仮想環境を作成。py3.7.4の所は、任意で構わないが以後はこの設定でやっていくので、各自で書き換えていただきたい。
  2. source py3.7.4/bin/activateを実行し、仮想環境を起動させる。
  • ビルド用にpipでパッケージインストール
  1. 下記のコマンドを1行ずつ実行し、インストールしていく。
pip install -U pip six numpy wheel setuptools mock
pip install -U keras_applications==1.0.6 --no-deps
pip install -U keras_preprocessing==1.0.5 --no-deps
  • Bazel0.24.1のインストール

1.sudo apt install pig-config zip g++ zliblg-dev unzipを実行し、前提となるシステムをインストールする。
2.https://github.com/bazelbuild/bazel/releases/tag/0.24.1
にアクセスし、bazel-0.24.1-installer-linux-x86_64.shをダウンロードする。
3.chmod +x bazel-0.24.1-installer-linux-x86_64.shを実行し、シェルスクリプトが動くようにする。
4../bazel-0.24.1-installer-linux-x86_64.sh --userを実行し、bazelをインストール。
5.vim ~/.bash_profileを実行し、以下の1行を入力し保存する。

export PATH="$PATH:$HOME/bin"

6.source .bash_profileを実行し、設定を読み込む。

  • tensorflowのソースファイルをビルド
  1. git clone https://github.com/tensorflow/tensorflow.gitを実行し、ソースファイルを取り込む。
  2. cd tensorflowでソースファイルの階層に移動。
  3. git checkout r1.14を実行し、1.14系のブランチを作る。
  4. ./configureを実行し、質問に答えていく。基本的にYes or No形式の質問は全てNOを選択し、[Default ...]といったYes or Noとは違う形式の質問は、基本的にエンターキーを押す。
  5. bazel build --config opt --config monolithic tensorflow:libtensorflow_cc.soを入力し、ビルドを実行する。非常に長いので、時間がある時に行うべし。
  • 自分の導入したいプロジェクトで使う

1.git cloneした時にできたtensorflowのフォルダをそのまま自分が使用しているプロジェクトフォルダにペーストする。
2.CMakeLists.txtとディレクトリ構造を下記に示す。

ディレクトリ構造

Projectファイル(今回は、test)
|-CMakeLists.txt
|-main.cpp
|-tensorflow(git cloneしてできたフォルダそのまま持ってくる)

CMakeLists.txt の中身

cmake_minimum_required(VERSION 3.14)
project(test)

#---追記---
set(CMAKE_CXX_FLAGS "$(CMAKE_CXX_FLAGS) -g -std=c++11 -W")

#bazel-bin,bazel-genfilesは、シンボリックなので、移行の際は注意
link_directories(./tensorflow/bazel-bin/tensorflow)

include_directories(
    ./tensorflow/
    ./tensorflow/bazel-genfiles
    ./tensorflow/bazel-bin/tensorflow
)
#---追記end---

add_executable(test main.cpp)

#---追記---
target_link_libraries(test tensorflow_cc tensorflow_framework)
#---追記end---

以上で、自分はJetbrains社製のCLionでやりましたが、ビルドが無事通りました。

ちなみに、ソースファイルの中身は


#include <stdio.h>
#include "tensorflow/tensorflow/c/c_api.h"

int main() {
   printf("Hello from Tensorflow C library version %s\n", TF_Version());
   return 0;
}

です。

2
1
0

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
2
1