TensorflowでGPUが使えるかを確認するコマンド
GPU搭載のPCでtensorflowを使うときに、ちゃんとGPUを使うことができるのかを確認したいときがあると思う。
このときは、以下のコマンドを実行すれば確認できる。
python # かpython3
まず、ターミナルでpython
を入力してpythonの環境に入る。
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
これで色々と文字が出力されてきたらGPUが使えるということが分かる。