tensorflow使用gpu加速的代码例子

以下是一个使用TensorFlow和GPU加速的代码示例：

import tensorflow as tf

# 检查是否有可用的GPU
if tf.test.is_gpu_available():
  print('GPU is available')
else:
  print('GPU is not available')

# 创建一个张量
a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')

# 创建一个变量
b = tf.Variable(tf.random_normal([3, 2], stddev=0.1), name='b')

# 创建一个操作
c = tf.matmul(a, b)

# 启动TensorFlow会话并运行操作
with tf.Session() as sess:
  sess.run(tf.global_variables_initializer())
  print(sess.run(c))

在上面的代码中，我们首先检查是否有可用的GPU，然后创建一个包含随机数的变量和一个包含常量值的张量。接下来，我们创建一个操作，该操作将变量和张量相乘，然后我们使用TensorFlow会话来运行操作。如果我们的计算机具有可用的GPU，TensorFlow将使用GPU来加速计算，否则将使用CPU。