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

Modelando relaciones en UML, un acercamiento a las Asociaciones

Utilizando Intents implícitos para crear actividades

Secuencias…Modelado indispensable