tensorflow2中如何设置GPU的使用

import tensorflow as tf

gpus = tf.config.list_physical_devices("GPU")
print(gpus)
if gpus:
    gpu0 = gpus[0] #如果有多个GPU，仅使用第0个GPU
    tf.config.experimental.set_memory_growth(gpu0, True) #设置GPU显存用量按需使用
    
    tf.config.set_visible_devices([gpu0],"GPU")