【Python】TensorflowでGPUが使えるかを確認するコマンド

TensorflowでGPUが使えるかを確認するコマンド

GPU搭載のPCでtensorflowを使うときに、ちゃんとGPUを使うことができるのかを確認したいときがあると思う。

このときは、以下のコマンドを実行すれば確認できる。

python # かpython3 

まず、ターミナルでpythonを入力してpythonの環境に入る。

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

これで色々と文字が出力されてきたらGPUが使えるということが分かる。

タイトルとURLをコピーしました