tensorflow 设置调用gpu

可以使用以下代码设置在TensorFlow中调用GPU：

import tensorflow as tf

# 设置TensorFlow使用GPU
physical_devices = tf.config.list_physical_devices('GPU')
tf.config.experimental.set_memory_growth(physical_devices[0], True)

这将启用TensorFlow在GPU上执行计算。