tensorflow-gpu

问题 I built the gpu version of the docker image https://github.com/floydhub/dl-docker with keras version 2.0.0 and tensorflow version 0.12.1. I then ran the mnist tutorial https://github.com/fchollet/keras/blob/master/examples/mnist_cnn.py but realized that keras is not using GPU. Below is the output that I have root@b79b8a57fb1f:~/sharedfolder# python test.py Using TensorFlow backend. Downloading data from https://s3.amazonaws.com/img-datasets/mnist.npz x_train shape: (60000, 28, 28, 1) 60000

问题 As per TensorFlow documentation , the prefetch and map methods of tf.contrib.data.Dataset class, both have a parameter called buffer_size . For prefetch method, the parameter is known as buffer_size and according to documentation : buffer_size: A tf.int64 scalar tf.Tensor, representing the maximum number elements that will be buffered when prefetching. For the map method, the parameter is known as output_buffer_size and according to documentation : output_buffer_size: (Optional.) A tf.int64