Skip to content

To verify that TensorFlow is using the GPU properly

Notifications You must be signed in to change notification settings

Summer498/HelloGPU

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 

Repository files navigation

HelloGPU

これは何?

  • Nvidia 製 GPU 持ちのマシンで TensorFlow がきちんと GPU を使っているのか確認するためのプログラム
  • .py 版と .ipynb 版がある

使い方

Windows

git clone https://github.com/Summer498/HelloGPU/
python HelloGPU\HelloGPU.py

Linux & Mac

git clone https://github.com/Summer498/HelloGPU/
python HelloGPU/HelloGPU.py

結果の見方

1.お使いのGPUの名前 (GeForce とか) がログとして出てくれば少なくともGPUを認識している

2.タスクマネージャーでGPUの使用量が上がっていることを確認

  • HelloGPU[.py|.ipynb] 内のバッチサイズを大きめにとると GPU の使用率が上がるので確認しやすい

参照

@nemutas, "Anaconda + Keras でGPUを使用する環境を構築する", Qiita

About

To verify that TensorFlow is using the GPU properly

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published