[Python] GPU 속도측정

[GPU 속도 측정]

GPUtest.py

tf.constant로 되어있는걸 placeholder로 바꾸고 sess.run할때 feed_dict하도록 바꾸기.

=> GPUtest-revised.py

같은 조건에서 cpu-cpu 8분 gpu-cpu4 분

GPUtest-revised.py

placeholder로 바꾸면 6초 내외

변수로 지정하고 돌렸더니 6초 정도 나옴.

나중에 GPU 셋팅할 때 정상적으로 작동하는지 도움이 되는 간단한 코드입니다. 

물론 CNN도 해보고 정확도도 해봐야하며, cudnn 5.1이 cudnn 5.0보다 2.7배 차이도 나기 때문에 5.1도 체크하셔야 합니다.

- 내 LG Gram Notebook에서 Docker Toolbox로 해보니 33분이 걸렸음.