cuda and cudnn version

[mudong0419]

Thanks for your work. What are versions of your cuda and cudnn?

[golbin]

I use cuda 8 and cudnn 5, 6 but I think it doesn’t care.

[ahnHeejune]

Thanks for your sharing the code. But, your code does not work on my machine with 3 Tian Xp GPUs. I spent some time and I found it run with " optimizer = tf.train.AdamOptimizer(learning_rate).minimize( loss, colocate_gradients_with_ops=False)" not with True. What do you think about this?