- Nvidia 製 GPU 持ちのマシンで TensorFlow がきちんと GPU を使っているのか確認するためのプログラム
- .py 版と .ipynb 版がある
git clone https://github.com/Summer498/HelloGPU/
python HelloGPU\HelloGPU.py
git clone https://github.com/Summer498/HelloGPU/
python HelloGPU/HelloGPU.py
1.お使いのGPUの名前 (GeForce とか) がログとして出てくれば少なくともGPUを認識している
- 出てこなければ認識すらしていない.
- CUDA, CuDNN, TensorFlow のバージョン合わせをやり直す
2.タスクマネージャーでGPUの使用量が上がっていることを確認
- HelloGPU[.py|.ipynb] 内のバッチサイズを大きめにとると GPU の使用率が上がるので確認しやすい