如何使用gpu运行python程序，请给出python代码

使用GPU运行Python程序需要使用GPU加速计算库，例如TensorFlow或PyTorch。以下是使用TensorFlow在GPU上运行Python程序的示例代码：

import tensorflow as tf

# 检查是否有可用的GPU
if not tf.test.gpu_device_name():
    print('No GPU found!')
else:
    print('GPU found!')

# 在GPU上运行TensorFlow会话
with tf.device('/gpu:0'):
    a = tf.constant([1.0, 2.0, 3.0, 4.0], shape=[2, 2], name='a')
    b = tf.constant([4.0, 3.0, 2.0, 1.0], shape=[2, 2], name='b')
    c = tf.matmul(a, b)

# 打印结果
print(c)

在上面的示例中，我们首先检查是否有可用的GPU，然后在GPU上运行TensorFlow会话并执行矩阵乘法。最后，我们打印结果。