質問をすることでしか得られない、回答やアドバイスがある。

15分調べてもわからないことは、質問しよう!

ただいまの
回答率

88.77%

tensorflowでobject detectionのモデルの学習プログラムを動かしてるのですが、うまくGPUを使って学習ができません。

解決済

回答 1

投稿

  • 評価
  • クリップ 0
  • VIEW 1,099

gjmwtaptm

score 12

 環境

AWSのDeep learning AMIを使ってEC2上で動かしています。

source activate tensorflow_p36し、
pip install -U tensorflowした状態で、

実行コマンドは下記です。(X)

python object_detection/model_main.py --pipeline_config_path=configs/faster_rcnn_resnet101_voc07.config --model_dir=data/output-model/ --num_train_steps=50000 --sample_1_of_n_eval_examples=1 --alsologtostderr

https://github.com/tensorflow/models/blob/master/research/object_detection

 問題

以下をobject_detection/model_main.pyに追加して、Xを実行すると

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

2018-11-09 18:12:12.582533: I tensorflow/core/platform/cpu_feature_guard.cc:141] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 FMA
Device mapping:
/job:localhost/replica:0/task:0/device:XLA_CPU:0 -> device: XLA_CPU device
2018-11-09 18:12:12.591782: I tensorflow/core/common_runtime/direct_session.cc:307] Device mapping:
/job:localhost/replica:0/task:0/device:XLA_CPU:0 -> device: XLA_CPU device

とログに出てきて、GPUが使えていないようです。

 やったこと

調べてみると、pip install tensorflow-gpu
を実行すればGPU使って学習できるいう記事を見つけたので、

https://github.com/tensorflow/models/issues/3366

以下を実行してからコードXを実行しました。

pip uninstall tensorflow
pip uninstall tensorflow-gpu
pip install -U tensorflow-gpu

しかし下記のエラーがでます。(少なくとも確認した限りバージョン:1.6,1.8,1.11で発生)
ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory

これの解決策としては、tensorflow-gpu:1.4にするといいようです。
https://github.com/tensorflow/tensorflow/issues/15604

そこで、下記を実行してコードXを実行しました。

pip uninstall tensorflow
pip uninstall tensorflow-gpu
pip install --upgrade tensorflow-gpu==1.4

しかしAttributeError: module 'tensorflow.python.keras' has no attribute 'Model'
というエラーが出ます。

これの解決策としては、pip install -U tensorflowを実行することが提案されており、無限ループに陥ってしまっています。
https://github.com/tensorflow/tensorflow/issues/21927

  • 気になる質問をクリップする

    クリップした質問は、後からいつでもマイページで確認できます。

    またクリップした質問に回答があった際、通知やメールを受け取ることができます。

    クリップを取り消します

  • 良い質問の評価を上げる

    以下のような質問は評価を上げましょう

    • 質問内容が明確
    • 自分も答えを知りたい
    • 質問者以外のユーザにも役立つ

    評価が高い質問は、TOPページの「注目」タブのフィードに表示されやすくなります。

    質問の評価を上げたことを取り消します

  • 評価を下げられる数の上限に達しました

    評価を下げることができません

    • 1日5回まで評価を下げられます
    • 1日に1ユーザに対して2回まで評価を下げられます

    質問の評価を下げる

    teratailでは下記のような質問を「具体的に困っていることがない質問」、「サイトポリシーに違反する質問」と定義し、推奨していません。

    • プログラミングに関係のない質問
    • やってほしいことだけを記載した丸投げの質問
    • 問題・課題が含まれていない質問
    • 意図的に内容が抹消された質問
    • 過去に投稿した質問と同じ内容の質問
    • 広告と受け取られるような投稿

    評価が下がると、TOPページの「アクティブ」「注目」タブのフィードに表示されにくくなります。

    質問の評価を下げたことを取り消します

    この機能は開放されていません

    評価を下げる条件を満たしてません

    評価を下げる理由を選択してください

    詳細な説明はこちら

    上記に当てはまらず、質問内容が明確になっていない質問には「情報の追加・修正依頼」機能からコメントをしてください。

    質問の評価を下げる機能の利用条件

    この機能を利用するためには、以下の事項を行う必要があります。

回答 1

check解決した方法

0

source activate mxnet_p36
pip install -U tensorflow-gpu
でコマンドXでうまくいきました。

投稿

  • 回答の評価を上げる

    以下のような回答は評価を上げましょう

    • 正しい回答
    • わかりやすい回答
    • ためになる回答

    評価が高い回答ほどページの上位に表示されます。

  • 回答の評価を下げる

    下記のような回答は推奨されていません。

    • 間違っている回答
    • 質問の回答になっていない投稿
    • スパムや攻撃的な表現を用いた投稿

    評価を下げる際はその理由を明確に伝え、適切な回答に修正してもらいましょう。

15分調べてもわからないことは、teratailで質問しよう!

  • ただいまの回答率 88.77%
  • 質問をまとめることで、思考を整理して素早く解決
  • テンプレート機能で、簡単に質問をまとめられる

関連した質問

同じタグがついた質問を見る

  • トップ
  • Python 3.xに関する質問
  • tensorflowでobject detectionのモデルの学習プログラムを動かしてるのですが、うまくGPUを使って学習ができません。