Python 3.x tensorflow不使用gpu-素数程序

我已经在我的笔记本电脑（GTX 1060-6GB）中安装了tensorflow gpu，我正在使用tensorflow开发素数生成器程序。下面是我用来计算素数的示例代码：

def prime_tf_graf(max_count):
    # matrix version
    with tf.device('/gpu:0'):
        data = tf.range(2, max_count, dtype=tf.int32)
        Y, X = tf.meshgrid(data, data); Z = X%Y
    with tf.device('/cpu:0'):
        loer = tf.matrix_band_part(Z, -1, 0) - tf.matrix_band_part(Z,  0, 0)
    with tf.device('/gpu:0'):
        loer = tf.cast(tf.not_equal(loer, 0), tf.int32)
        sumr = tf.reduce_sum(loer, axis=1)
        nums = data - 2
        rato = tf.cast(sumr/nums, tf.int32)
        indx =  tf.cast(tf.not_equal(rato, 0), tf.int32)
    return tf.reshape(tf.where(indx), [-1]) + 2

def prime_tf(max_count):
    graf = prime_tf_graf(max_count)
    sess = tf.Session(config=tf.ConfigProto(allow_soft_placement=True))
    prim = sess.run(graf)
    return prim

当我运行这段代码时，我的CPU和RAM只会被使用，而我的GPU不会被使用。（仅分配专用内存）。测试运行：

prime_tf（tf.常量（20000））

以下是我的系统用法：

---

它表明tensorflow只使用我的CPU和RAM，而不是GPU。

使用变量而不是常量。.如何以及在何处使用？