Tech

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

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


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"
memory_limit: 268435456

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

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

Kerasに関する基礎知識を身に着ける際は以下の書籍がおすすめだ。