tensorflowでGPUが使われているか確認する
tensorflow-gpuのインストールは
を参照してください。アメリカ語が読めない人は機械学習は無理です諦めてください。
で、正しく使われているか確認したい時は
>>> 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())
って方法が出てくるけど紹介したヤツのほうが簡単だと思う。