Saturday, August 6, 2016

TensorFlow vs Theano, CPU Perf, on an old laptop

Laptop: Aspire 4755G (Intel i5-2450M)
OS: Ubuntu 14.04

Theano Version: 0.8.2
TensorFlow Version: 0.10.0rc0

This laptop has a GPU but it's compute capability is not high enough to meet TensorFlow's requirements.

The test: 5 optimizing minibatch iterations with a small "modern neural net" which classifies MNIST digits.
Original Theano version:
https://github.com/Newmu/Theano-Tutorials/blob/master/4_modern_net.py
TensorFlow version:
https://github.com/nlintz/TensorFlow-Tutorials/blob/master/04_modern_net.py

I changed the number of iterations to 5.

Results:
Tensorflow took 80 seconds.
Theano took 180 seconds.


Bonus: Theano using the laptop's GPU took 37 seconds (NOT including the initialization time.)

No comments :

Post a Comment