THUMBS SHIFT→

このブログは主に親指シフトを用いて書かれています

tensorflowでGPUが使われているか確認する

tensorflow-gpuのインストールは

GPU support  |  TensorFlow

を参照してください。アメリカ語が読めない人は機械学習は無理です諦めてください。

で、正しく使われているか確認したい時は

>>> import tensorflow as tf
>>> tf.test.is_gpu_available()
True

で確認できる。

参考: https://www.tensorflow.org/api_docs/python/tf/test/is_gpu_available

検索すると

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

って方法が出てくるけど紹介したヤツのほうが簡単だと思う。