speed benchmark for testing tensorflow install

Question

I\'m doubting whether tensorflow is correctly configured on my gpu box, since it\'s about 100x slower per iteration to train a simple linear regression model (batchsize = 32, 15

Answer 1

Try tensorflow/tensorflow/models/image/mnist/convolutional.py, that'll print per-step timing.

On Tesla K40c that should get about 16 ms per step, while about 120 ms for CPU-only on my 3 year old machine

---

Edit: This moved to the models repositories: https://github.com/tensorflow/models/blob/master/tutorials/image/mnist/convolutional.py.

The convolutional.py file is now at models/tutorials/image/mnist/convolutional.py