CV・NLPハマりどころメモ

画像認識と自然言語処理を研究する中でうまくいかなかったことと、その対策をまとめる自分用メモが中心。

KerasがGPU版のTensorflowを使っているかを確認する方法[Keras][Tensorflow]

以下手順

>>> from tensorflow.python.client import device_lib
>>> print(device_lib.list_local_devices()) # このコマンドの出力でGPUかCPUかが分かる
2018-09-11 07:13:29.668907: I tensorflow/core/platform/cpu_feature_guard.cc:140] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 FMA
[name: "/device:CPU:0"
device_type: "CPU" # CPUだったorz...
memory_limit: 268435456

Nvidia-docker(https://github.com/NVIDIA/nvidia-docker)で固有表現抽出のコンテナをインストールしたのだが、中で動いているkerasがGPUで動いているのかどうかが分からなかった為、本コマンドを使って調べたのだ。

結果、CPUで動いていることが分かったので、もう一度コンテナを立てなおさねば。