当我与“ tf.device（'/ device：GPU：0'）”一起使用时，为什么代码仍不使用GPU

Question

我只是使用TensorFlow训练模型。但是，当我在tf.device（'/ device：GPU：0'）中使用命令：with时，代码仍然没有使用GPU进行计算。

您能帮我吗？

谢谢！

我已经在tf.device（'device：GPU：0'）中使用了命令，但是代码仍在CPU上运行。

###========================== DEFINE MODEL ============================###
## train inference
# create placeholder variables which serve as the inputs to graphs

with tf.device('/device:GPU:0 '):

    t_image = tf.placeholder('float32', [batch_size, block_size, 
     block_size, input_depth],
                             name='t_image')  # input LR image

      im_size_HR = int(block_size * ratio)