如何使用GPU在批处理模式下使用张量流预训模型（即初始v3）？

Question

我想在张量流中使用初始v3模型进行特征提取。但是我使用的图像数量很多，因此运行需要很长时间。所以，我将使用GPU。我已经正确安装了Cuda 7.5和cuDnn。 我在CPU模式下使用以下代码作为一个图像：

with tf.Session as sess:
    softmax_tensor =sess.graph.get_tensor_by_name('pool_3:0')
    feat_vect = numpy.squeeze(sess.run(softmax_tensor,{'DecodeJpeg:0':       in_image}))

所以，我的问题是我应该如何更改我的代码，以便我可以通过GPU将它用于多批次？