Utilizando TensorFlow 2 preview, una simple comparación entre el rendimiento CPU vs GPU

Utilizando tf 2 en modo eager , un método mucho más afín a coder/desarrollador, intenté realizar un procedimiento simple para multiplicar matrices y explorar el rendimiento entre la ejecución en CPU y GPU.
Utilizando una multiplicación de 2000x2000 e iterando en 100 steps, tenemos estos valores de comparación en segundos.


*para la ejecución utilicé una instancia VM provista por Google Cloud, la imagen con soporte GPU, la m21.


La rutina es muy simple, puede ver su implementación aquí desde este proyecto GitHub. device_check.py

Comentarios

Entradas más populares de este blog

Regresión lineal simple en #TensorFlow, aprendizaje automático para mayor exactitud

Utilizando Intents implícitos para crear actividades

Humedad y temperatura desde #Arduino a #Firebase